Base package for implementing K-Fold experiments

Install

npm install kfold-experiment

Usage

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()

Run tests

npm run test

Author

👤 Marco Cardoso

• Github: @MarcoABCardoso
• LinkedIn: @marco-cardoso

Show your support

Give a ⭐️ if this project helped you!