Compare commits

...

1 Commits

Author SHA1 Message Date
rmschubert
7ae7578845 fix: update scikit-learn dependency 2022-12-05 16:31:33 +01:00

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 = [