UNPKG

@galihru/tvmai

Version:

Training and Evaluation AI Model Recommendation Engine for WebNN

31 lines (30 loc) 991 B
import { loadImageDataset } from './loaders/imageLoader'; import { loadPDFDataset } from './loaders/pdfLoader'; import { loadCSVDataset } from './loaders/csvLoader'; import { analyzeDataset } from './core/datasetAnalyzer'; import { recommendModel } from './core/modelRecommender'; export const loadDataset = async (file) => { if (file.name.endsWith('.zip')) { if (file.name.toLowerCase().includes('image')) { return loadImageDataset(file); } else if (file.name.toLowerCase().includes('pdf')) { return loadPDFDataset(file); } else { try { return await loadImageDataset(file); } catch (e) { return await loadPDFDataset(file); } } } else if (file.name.endsWith('.csv')) { return loadCSVDataset(file); } else { throw new Error('Unsupported file format'); } }; export { analyzeDataset, recommendModel };