techavenger123 commited on
Commit ·
a949781
1
Parent(s): 7a8e988
added best config
Browse files
app.py
CHANGED
|
@@ -52,10 +52,10 @@ FIXED_PARAMS = dict(
|
|
| 52 |
|
| 53 |
BEST_CONFIG = {
|
| 54 |
"learning_rate": 0.05,
|
| 55 |
-
"n_estimators":
|
| 56 |
-
"train_ratio": 0.
|
| 57 |
-
"val_ratio": 0.
|
| 58 |
-
"test_ratio": 0.
|
| 59 |
}
|
| 60 |
|
| 61 |
EQUIPMENT_OPTIONS = ["pump", "compressor", "motor", "valve", "sensor"]
|
|
|
|
| 52 |
|
| 53 |
BEST_CONFIG = {
|
| 54 |
"learning_rate": 0.05,
|
| 55 |
+
"n_estimators": 10,
|
| 56 |
+
"train_ratio": 0.50,
|
| 57 |
+
"val_ratio": 0.20,
|
| 58 |
+
"test_ratio": 0.30,
|
| 59 |
}
|
| 60 |
|
| 61 |
EQUIPMENT_OPTIONS = ["pump", "compressor", "motor", "valve", "sensor"]
|