techavenger123 commited on
Commit ·
2f37dc9
1
Parent(s): 88122df
Project Structure File.
Browse files- Dockerfile +1 -0
- app.py +4 -4
- app3.py +1 -1
Dockerfile
CHANGED
|
@@ -13,5 +13,6 @@ COPY --chown=user . /app
|
|
| 13 |
|
| 14 |
USER user
|
| 15 |
|
|
|
|
| 16 |
EXPOSE 7860
|
| 17 |
CMD ["python", "-m", "gunicorn", "--bind", "0.0.0.0:7860", "--workers", "2", "--timeout", "120", "app3:app"]
|
|
|
|
| 13 |
|
| 14 |
USER user
|
| 15 |
|
| 16 |
+
|
| 17 |
EXPOSE 7860
|
| 18 |
CMD ["python", "-m", "gunicorn", "--bind", "0.0.0.0:7860", "--workers", "2", "--timeout", "120", "app3:app"]
|
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": 165,
|
| 56 |
+
"train_ratio": 0.80,
|
| 57 |
+
"val_ratio": 0.10,
|
| 58 |
+
"test_ratio": 0.10,
|
| 59 |
}
|
| 60 |
|
| 61 |
EQUIPMENT_OPTIONS = ["pump", "compressor", "motor", "valve", "sensor"]
|
app3.py
CHANGED
|
@@ -49,7 +49,7 @@ LGBM_PARAMS = dict(
|
|
| 49 |
random_state=RANDOM_STATE,
|
| 50 |
verbose=-1,
|
| 51 |
learning_rate=0.05,
|
| 52 |
-
n_estimators=
|
| 53 |
)
|
| 54 |
|
| 55 |
RF_PARAMS = dict(
|
|
|
|
| 49 |
random_state=RANDOM_STATE,
|
| 50 |
verbose=-1,
|
| 51 |
learning_rate=0.05,
|
| 52 |
+
n_estimators=165,
|
| 53 |
)
|
| 54 |
|
| 55 |
RF_PARAMS = dict(
|