Updated mlflow and scikit-learn (#3130)
* Updated mlflow and scikit-learn * Changed to use latest sklearn 1.0 curated environment * Update Pytorch environment
This commit is contained in:
Родитель
a946223041
Коммит
35455af3c8
|
@ -2,7 +2,7 @@
|
|||
numpy==1.22
|
||||
scipy==1.7.1
|
||||
pandas==1.3.0
|
||||
scikit-learn==0.24.2
|
||||
scikit-learn==1.2.2
|
||||
adlfs==2021.9.1
|
||||
fsspec==2021.8.1
|
||||
|
||||
|
@ -11,8 +11,8 @@ xgboost==1.4.2
|
|||
lightgbm==3.2.1
|
||||
|
||||
# tracking
|
||||
mlflow==1.20.2
|
||||
azureml-mlflow==1.34.0
|
||||
mlflow==2.2.1
|
||||
azureml-mlflow==1.55.0
|
||||
|
||||
# visualization
|
||||
matplotlib==3.4.3
|
||||
|
|
|
@ -241,7 +241,7 @@
|
|||
" ),\n",
|
||||
" },\n",
|
||||
" # we're using a curated environment\n",
|
||||
" environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu:1\",\n",
|
||||
" environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu@latest\",\n",
|
||||
" compute=\"cpu-cluster\"\n",
|
||||
" if (cpu_cluster)\n",
|
||||
" else None, # No compute needs to be passed to use serverless\n",
|
||||
|
@ -325,7 +325,7 @@
|
|||
" ),\n",
|
||||
" },\n",
|
||||
" # we're using a curated environment\n",
|
||||
" environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu:1\",\n",
|
||||
" environment=\"AzureML-sklearn-1.0-ubuntu20.04-py38-cpu@latest\",\n",
|
||||
" compute=\"cpu-cluster-lg\",\n",
|
||||
")\n",
|
||||
"\n",
|
||||
|
@ -412,7 +412,7 @@
|
|||
" # \"register_model_as\": \"places_dev\",\n",
|
||||
" \"enable_profiling\": False,\n",
|
||||
" },\n",
|
||||
" environment=\"AzureML-pytorch-1.10-ubuntu18.04-py38-cuda11-gpu@latest\",\n",
|
||||
" environment=\"AzureML-acpt-pytorch-1.13-cuda11.7@latest\",\n",
|
||||
" compute=\"gpu-cluster\"\n",
|
||||
" if (gpu_cluster)\n",
|
||||
" else None, # No compute needs to be passed to use serverless\n",
|
||||
|
|
|
@ -4,7 +4,7 @@ torch==1.10.2
|
|||
|
||||
# for metrics reporting/plotting
|
||||
azureml-mlflow==1.40.0
|
||||
mlflow==1.23.1
|
||||
mlflow==2.2.1
|
||||
matplotlib==3.5.1
|
||||
|
||||
# to run the notebooks locally
|
||||
|
|
Загрузка…
Ссылка в новой задаче