Marcao-kfold NPM

Base package for implementing K-Fold experiments

Install

npm install marcao-kfold

Usage

const Experiment = require('marcao-kfold')

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