22 lines
564 B
YAML
22 lines
564 B
YAML
name: linear_regression
|
|
path: linear_regression_model_wrapper
|
|
class: LinearRegressionModelWrapper
|
|
runtime: basic
|
|
requirements:
|
|
- pandas==2.3.3
|
|
- scikit-learn==1.8.0
|
|
- cloudpickle==3.1.2
|
|
- mlflow==3.8.1
|
|
- git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.11.0
|
|
- git+ssh://git@github.com/Aignosi/sientia-model-library.git@0.14.0
|
|
version: 1.2.0
|
|
type: model
|
|
model:
|
|
class: LinearRegressionModel
|
|
external: false
|
|
path: linear_regression_model
|
|
data_model:
|
|
class: DataPreprocessor
|
|
external: true
|
|
path: sientia_model.preprocessing.data_preprocessor
|