File tree 1 file changed +22
-0
lines changed
tutorials/get-started-notebooks
1 file changed +22
-0
lines changed Original file line number Diff line number Diff line change 252
252
" ##########################\n " ,
253
253
" # Registering the model to the workspace\n " ,
254
254
" print(\" Registering the model via MLFlow\" )\n " ,
255
+ " \n " ,
256
+ " # pin numpy\n " ,
257
+ " conda_env = {\n " ,
258
+ " 'name': 'mlflow-env',\n " ,
259
+ " 'channels': ['conda-forge'],\n " ,
260
+ " 'dependencies': [\n " ,
261
+ " 'python=3.10.15',\n " ,
262
+ " 'pip<=21.3.1',\n " ,
263
+ " {\n " ,
264
+ " 'pip': [\n " ,
265
+ " 'mlflow==2.17.0',\n " ,
266
+ " 'cloudpickle==2.2.1',\n " ,
267
+ " 'pandas==1.5.3',\n " ,
268
+ " 'psutil==5.8.0',\n " ,
269
+ " 'scikit-learn==1.5.2',\n " ,
270
+ " 'numpy==1.26.4',\n " ,
271
+ " ]\n " ,
272
+ " }\n " ,
273
+ " ],\n " ,
274
+ " }\n " ,
275
+ " \n " ,
255
276
" mlflow.sklearn.log_model(\n " ,
256
277
" sk_model=clf,\n " ,
257
278
" registered_model_name=args.registered_model_name,\n " ,
258
279
" artifact_path=args.registered_model_name,\n " ,
280
+ " conda_env=conda_env,\n " ,
259
281
" )\n " ,
260
282
" \n " ,
261
283
" # Saving the model to a file\n " ,
You can’t perform that action at this time.
0 commit comments