Component: TensorFlow Machine Learning

From MaxFlow Wiki
Jump to navigation Jump to search

Purpose

This component is a simple interface for the machine learning library TensorFlow. There are two components one for training the model, and the other for predicting.

How to use

Use the training to obtain a trained model and save it to a file. Then use the saved model for the the predicting component.

  • TensorFlow Model Training: takes in input sets (x_train) and a set of output set (y_train) and generates a trained model.
  • TensorFlow Prediction: takes in a trained model (saved by the training component) and an input set and generate the index and accuracy of the output.