UBC-MDS / GoT-fatality-prediction

https://ubc-mds.github.io/GoT-fatality-prediction/
Other
0 stars 1 forks source link

added script for fitting and eval model and updated dockerfile #47

Closed 786213750 closed 7 months ago

786213750 commented 7 months ago
# train model, save model object and random search object
python scripts/fit_got_fatality_classifier.py \
 --training-data data/processed/train.csv \
--model-to results/models/

# evaluate model on test data and save results
python scripts/evaluate_got_fatality_predictor.py \
--test-data data/processed/test.csv \
--models-from results/models/ \
--results-to results/