25 lines
526 B
YAML
25 lines
526 B
YAML
name: xgboost
|
|
path: xgboost_model_wrapper
|
|
class: XGBoostWrapper
|
|
runtime: xgboost
|
|
requirements:
|
|
- mlflow==3.8.1
|
|
- pathspec==1.0.4
|
|
- pandas==2.3.3
|
|
- xgboost ==3.1.2
|
|
- cloudpickle==3.1.2
|
|
- rich==14.3.2
|
|
- joblib==1.5.3
|
|
- git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.11.0
|
|
- git+ssh://git@github.com/Aignosi/sientia-model-library.git@0.8.1
|
|
version: 1.1.1
|
|
type: model
|
|
model:
|
|
external: true
|
|
class: Booster
|
|
path: xgboost
|
|
data_model:
|
|
class: TimeSeriesPreprocessor
|
|
external: false
|
|
path: xgboost_data_model
|