Skip to content

CytoClassifier Machine Learning

David Dao edited this page Apr 18, 2018 · 8 revisions

How to hack machine learning in JS

Hey guys, this section will point you to resources for tensorflow.js - which will be our main ml toolkit!

TensorFlow in JS

Designing an API for ML kits

We aim to implement all algorithms with tfjs layers, while exposing a keras-like API:

  • .fit() for training
  • .predict() for inference

More infos here

Goals for the Hack

JS implementations

  • Implement Gradient Boosting Trees
  • Implement Hyperparameter/Model Search
  • Implement Confusion Matrices / Cross Validation

Pretrained Model (from python in js)

@daviddao

Cool examples

Yolo9000 in the browser Link