Skip to content

Commit

Permalink
254 add CLI functions for ML models (#262)
Browse files Browse the repository at this point in the history
Added:

7 CLI functions:
- random_forest_classifier_train_cli
- random_forest_regressor_train_cli
- gradient_boosting_classifier_train_cli
- gradient_boosting_regressor_train_cli
- logistic_regression_train_cli
- evaluate_trained_model_cli
- predict_with_trained_model_cli

- Notebook with examples of LR, RF and GB use

- machine_learning_general.py. module. Has all general ML functions and utilities. reshape_predictions and prepare_data_for_ml are new additions
  • Loading branch information
nmaarnio authored Jan 10, 2024
1 parent 253bd89 commit a715cb4
Show file tree
Hide file tree
Showing 10 changed files with 1,366 additions and 26 deletions.
3 changes: 3 additions & 0 deletions docs/prediction/machine_learning_general.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Logistic regression

::: eis_toolkit.prediction.machine_learning_general
Loading

0 comments on commit a715cb4

Please sign in to comment.