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:
jeff-shepherd 2024-04-29 18:20:54 -07:00 коммит произвёл GitHub
Родитель a946223041
Коммит 35455af3c8
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
3 изменённых файлов: 7 добавлений и 7 удалений

Просмотреть файл

@ -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