Skip to content

Commit d253567

Browse files
committed
Development
1 parent 2f186b7 commit d253567

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

mlOps/modelOps/data_science/nyc_taxi/train_register.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,9 @@
88
# Install pypi packages azureml-sdk[databricks], lightgbm, uszipcode
99
# The above will be automated in due course
1010

11+
12+
# https://learn.microsoft.com/en-us/azure/databricks/_extras/notebooks/source/machine-learning/automl-feature-store-example.html
13+
1114
# COMMAND ----------
1215

1316
from pyspark.sql import *
@@ -337,6 +340,9 @@ def train_model(
337340
)
338341

339342
#Save The Model
343+
344+
self.create_model_folder()
345+
340346
model_file_path = self.get_model_file_path("taxi_example_fare_packaged")
341347
print(f"ModelFilePath: {model_file_path}")
342348
joblib.dump(

0 commit comments

Comments
 (0)