MetOffice / MetOffice/XBTs_classification

Build tensorflow classifiers

Open
#39 0 comments 0 reactions 1 assignee Claimed by @stevehadd View on GitHub
libraries pipelines reproducible_research
Dominant language
Jupyter Notebook
Stars
4
Forks
2
PR merge metrics
No merged PRs in 30d

Description

We're currently running in to a limitatin of scikit-learn when we want to deploy the trained model. One can pickle the classifier object, but this is not a robust way to save the trained model.
https://scikit-learn.org/stable/modules/model_persistence.html

Other packages have better options for saving a trained model built in to the library, for example tensorflow. Tensor flow does not have a decision tree classifier, but does have a Gradient Boosted trees and random forest classifiers, which have shown similar results to DTs in scikit learn. We should thus try to move to using tensorflow (or other library potentially e.g. pyTorch, azure ML, AWS sagemaker etc.)

https://www.tensorflow.org/tutorials/estimator/boosted_trees
https://www.tensorflow.org/api_docs/python/tf/estimator/BoostedTreesClassifier
https://www.tensorflow.org/api_docs/python/tf/saved_model/save

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.