diff --git a/MLflow/MLProject.yml b/MLflow/MLProject.yml new file mode 100644 index 0000000..f0df069 --- /dev/null +++ b/MLflow/MLProject.yml @@ -0,0 +1,9 @@ +name: MobilePriceClassification + +conda_env: conda.yaml + +entry_points: + main: + parameters: + data_url: {type: str, default: ""} + command: "full.py" diff --git a/MLflow/conda.yml b/MLflow/conda.yml new file mode 100644 index 0000000..96c6e32 --- /dev/null +++ b/MLflow/conda.yml @@ -0,0 +1,12 @@ +name: mobile-price-classification +channels: + - defaults +dependencies: + - python=3.8 + - pandas + - torch + - scikit-learn + - mlflow + - pip + - pip: + - kaggle