Unverified Commit 79d3c492 authored by J-shang's avatar J-shang Committed by GitHub
Browse files

update sklearn version to 0.24.1 (#3407)

parent b0d62be8
......@@ -63,9 +63,9 @@ RUN python3 -m pip --no-cache-dir install torch==1.4.0
RUN python3 -m pip install torchvision==0.5.0
#
# sklearn 0.23.2
# sklearn 0.24.1
#
RUN python3 -m pip --no-cache-dir install scikit-learn==0.23.2
RUN python3 -m pip --no-cache-dir install scikit-learn==0.24.1
#
# pandas==0.23.4 lightgbm==2.2.2
......
......@@ -9,7 +9,7 @@ responses
schema
PythonWebHDFS
colorama
scikit-learn >= 0.23.2
scikit-learn >= 0.24.1
websockets
filelock
prettytable
......
......@@ -12,7 +12,7 @@ peewee
nbsphinx
schema
tensorboard
scikit-learn>=0.23.2
scikit-learn>=0.24.1
thop
colorama
pkginfo
......
......@@ -28,7 +28,7 @@ jobs:
- script: |
set -e
python3 -m pip install scikit-learn==0.23.2
python3 -m pip install scikit-learn==0.24.1
python3 -m pip install torchvision==0.7.0
python3 -m pip install torch==1.6.0
python3 -m pip install 'pytorch-lightning>=1.1.1,<1.2'
......
......@@ -25,7 +25,7 @@ jobs:
displayName: Install NNI
- script: |
python -m pip install scikit-learn==0.23.2
python -m pip install scikit-learn==0.24.1
python -m pip install keras==2.1.6
python -m pip install torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
python -m pip install 'pytorch-lightning>=1.1.1,<1.2'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment