Skip to content

Commit

Permalink
Update recipe example to mlflow 2.0 requirements
Browse files Browse the repository at this point in the history
Signed-off-by: Sunish Sheth <[email protected]>
  • Loading branch information
sunishsheth2009 committed Nov 15, 2022
1 parent 43c4d88 commit 0852303
Show file tree
Hide file tree
Showing 7 changed files with 11 additions and 13 deletions.
1 change: 0 additions & 1 deletion .github/workflows/integration.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@ jobs:
- name: Install dependencies
run: |
pip install -r ./requirements.txt
pip install mlflow@git+https://github.com/mlflow/mlflow@master
- name: Run example
run: |
cd regression
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ Your contribution to MLflow Recipes is greatly appreciated by the community!

## Installation instructions
To use MLflow Recipes in this example repository,
simply install the packages listed in the `requirements.txt` file. Note that `Python 3.8` or above is recommended.
simply install the packages listed in the `requirements.txt` file. Note that `Python 3.8` or above is required.
```
pip install requirements.txt
```
Expand Down
6 changes: 3 additions & 3 deletions classification/profiles/databricks.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -16,19 +16,19 @@ model_registry:
SPLIT_RATIOS: [0.75, 0.125, 0.125]

INGEST_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-classification-template#ingest-step
# For different options please read: https://github.com/mlflow/recipes-classification-template#ingest-step
using: csv
loader_method: load_file_as_dataframe
location:
- "/dbfs/databricks-datasets/wine-quality/winequality-white.csv"
- "/dbfs/databricks-datasets/wine-quality/winequality-red.csv"

INGEST_SCORING_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-classification-template#batch-scoring
# For different options please read: https://github.com/mlflow/recipes-classification-template#batch-scoring
using: parquet
location: ""

PREDICT_OUTPUT_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-classification-template#predict-step
# For different options please read: https://github.com/mlflow/recipes-classification-template#predict-step
using: table
location: "is_red_wine_prediction"
1 change: 0 additions & 1 deletion regression/notebooks/databricks.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
# COMMAND ----------

# MAGIC %pip install -r ../../requirements.txt
# MAGIC %pip install mlflow

# COMMAND ----------

Expand Down
6 changes: 3 additions & 3 deletions regression/profiles/databricks.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -16,16 +16,16 @@ model_registry:
SPLIT_RATIOS: [0.75, 0.125, 0.125]

INGEST_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-regression-template#ingest-step
# For different options please read: https://github.com/mlflow/recipes-regression-template#ingest-step
using: spark_sql
sql: SELECT * FROM delta.`dbfs:/databricks-datasets/nyctaxi-with-zipcodes/subsampled`

INGEST_SCORING_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-regression-template#batch-scoring
# For different options please read: https://github.com/mlflow/recipes-regression-template#batch-scoring
using: spark_sql
sql: SELECT * FROM delta.`dbfs:/databricks-datasets/nyctaxi-with-zipcodes/subsampled`

PREDICT_OUTPUT_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-regression-template#predict-step
# For different options please read: https://github.com/mlflow/recipes-regression-template#predict-step
using: table
location: "taxi_regression_batch_scoring"
6 changes: 3 additions & 3 deletions regression/profiles/local.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -15,16 +15,16 @@ model_registry:
SPLIT_RATIOS: [0.80, 0.10, 0.10]

INGEST_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-regression-template#ingest-step
# For different options please read: https://github.com/mlflow/recipes-regression-template#ingest-step
using: parquet
location: "./data/sample.parquet"

INGEST_SCORING_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-regression-template#batch-scoring
# For different options please read: https://github.com/mlflow/recipes-regression-template#batch-scoring
using: parquet
location: "./data/sample.parquet"

PREDICT_OUTPUT_CONFIG:
# For different options please read: https://github.com/mlflow/mlp-regression-template#predict-step
# For different options please read: https://github.com/mlflow/recipes-regression-template#predict-step
using: parquet
location: "./data/sample_output.parquet"
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
mlflow>=2.0.0rc0
mlflow>=2.0
scikit-learn>=1.1
ipykernel>=6.12
ipython>=7.32

0 comments on commit 0852303

Please sign in to comment.