@galihru/tvmai
Version:
Training and Evaluation AI Model Recommendation Engine for WebNN
23 lines (22 loc) • 827 B
JavaScript
import { calculateEntropy } from './mathUtils';
export const analyzeDataset = (metadata) => {
let analysis = {};
if (metadata.type === 'image') {
analysis.complexity = metadata.classes * metadata.avgResolution * metadata.channels;
analysis.dataType = 'image';
}
else if (metadata.type === 'tabular') {
const entropy = calculateEntropy(Object.values(metadata.classDistribution));
analysis.complexity = entropy * metadata.features;
analysis.dataType = 'tabular';
}
else if (metadata.type === 'text') {
analysis.complexity = metadata.avgLength * metadata.vocabSize;
analysis.dataType = 'text';
}
return {
...metadata,
...analysis,
recommendationKey: `${analysis.dataType}_${Math.round(analysis.complexity)}`
};
};