Some pretrained models of project PHBS_MLF_2019_Project .
Report | Code | Data & Model |
---|---|---|
📄 Report in PDF | Correlation and engineering | Description |
- | Cleaning | Train.csv |
- | Baselines and trees | Test.csv |
- | ANN | Models(You are here) |
-
Logistic Regession
-
Random Forest (only parameters uploaded due to file size)
-
GBDT / DART in LightGBM
-
GBDT in XGBoost
-
ANN (only parameters uploaded due to file size)
Type | Filename |
---|---|
Model dumped | {model} _ {datetime} . model |
e. g. LogisticRegression_04161442. model | |
Confusion matrix | {model} _ {datetime}. png |
e. g. LogisticRegression_04161442. png | |
Parameter | {model} _ {datetime} _ {accuracy} %. txt |
e. g. LogisticRegression_04161442_89. 04%. txt |
For ANN parameters: final_ann_model_mod.pt
.
This project is under MIT license.