/ml-for-engineers

Easy Machine-Learning using ludwig 👩‍🔬🎼

Primary LanguageShellMozilla Public License 2.0MPL-2.0

Easy Machine Learning With Ludwig

"Ludwig is a toolbox built on top of TensorFlow that allows to train and test deep learning models without the need to write code. http://uber.github.io/ludwig/"

Associated Blog Post

How to train

ludwig train -mdf model.yaml --data_csv yld.csv

How to predict

ludwig predict -m results/experiment_run_0/model --data_csv yld.csv
cat results_0/tags_predictions.csv

Prediction with text overlay on images

With 0 being the result folder number from the training run.

./predict.sh 0