feat(worker): add Hybrid_SpatioTemporal to valid models
Build and Push Docker Images / build-and-push (push) Successful in 15m45s
Details
Build and Push Docker Images / build-and-push (push) Successful in 15m45s
Details
This commit is contained in:
parent
cd9f6f210d
commit
f97bd14a7f
|
|
@ -26,7 +26,7 @@ STAGES = [
|
||||||
]
|
]
|
||||||
|
|
||||||
# Acceptable model names
|
# Acceptable model names
|
||||||
VALID_MODELS = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost"]
|
VALID_MODELS = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost", "Hybrid_SpatioTemporal"]
|
||||||
|
|
||||||
# Valid smoothing kernel sizes
|
# Valid smoothing kernel sizes
|
||||||
VALID_KERNEL_SIZES = [3, 5, 7]
|
VALID_KERNEL_SIZES = [3, 5, 7]
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue