Skip to content

A Node.JS Native add-ons to run Keras trained model.

License

Notifications You must be signed in to change notification settings

Aghabeiki/keras-predict

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

keras-predict

A Nodejs Native addons to run Keras ML model.

Run Keras ML model via frugally-deep as a native Nodejs Addons.

Usage:

  • first install frugally-deep.
  • compile your keras model :
python3 frugally-deep/keras_export/convert_model.py keras_model.h5 fdeep_model.json
  • path the size of your ML input and converted ML path to predict function.
const predict = require('keras-predict')
const path = require('path');
const mlModelPath = path.resolve(__dirname, "fdeep_model.json");
const testCase = [...];
predict.predict(120, mlModelPath, testCase, (err, predict) => {
       if (err) {
           console.error(err)
       } else {
           console.dir(predict.map(Math.round))
       }
   })
}
  • A working example avialable at example folder.
  • This module tested only on ArchLinux.

Amin Aghabeiki.