fix: update scikit-learn dependency

This commit is contained in:
rmschubert 2022-12-05 16:31:33 +01:00
parent 0649d5bb45
commit 7ae7578845

View File

@ -22,7 +22,7 @@ INSTALL_REQUIRES = [
"torch>=1.3.1", "torch>=1.3.1",
"torchvision>=0.7.4", "torchvision>=0.7.4",
"numpy>=1.9.1", "numpy>=1.9.1",
"sklearn", "scikit-learn",
"matplotlib", "matplotlib",
] ]
DATASETS = [ DATASETS = [