Base package for implementing K-Fold experiments
npm install kfold-experiment
const Experiment = require('kfold-experiment')
const experiment = new Experiment({
exportData: () => [], // Function that gets data, as an array of { input: any, class: string }
trainModel: (train) => {}, // Function that creates a new model, using an array of { input: any, class: string }
checkModelStatus: (model) => true, // Function that checks model status, resolves true when ready
predict: (model, input) => [], // Function that sends input to a model, resolves array of { class: string, confidence: number }
deleteModel: (model) => {}, // Function that deletes temporary model once done with testing
})
let results = await experiment.run()
npm run test
👤 Marco Cardoso
- Github: @MarcoABCardoso
- LinkedIn: @marco-cardoso
Give a ⭐️ if this project helped you!