kfold-experiment
TypeScript icon, indicating that this package has built-in type declarations

1.6.1 • Public • Published

kfold-experiment

Version License: ISC Coverage Status Node.js CI

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

Show your support

Give a ⭐️ if this project helped you!

Package Sidebar

Install

npm i kfold-experiment

Weekly Downloads

16

Version

1.6.1

License

ISC

Unpacked Size

22.8 kB

Total Files

10

Last publish

Collaborators

  • macardoso95