fix scikit-learn dep (#1103)

* fix scikit-learn dep

Signed-off-by: Sachidanand Alle <[email protected]>

* fix class name

Signed-off-by: Sachidanand Alle <[email protected]>

Signed-off-by: Sachidanand Alle <[email protected]>

Author: SachidanandAlle

requirements.txt

@@ -41,6 +41,6 @@ python-jose[cryptography]==3.3.0
 bcrypt==3.2.2
 shapely==1.8.2
 requests==2.28.1
-sklearn
+scikit-learn
 
 #sudo apt-get install openslide-tools -y

sample-apps/pathology/lib/handlers.py

@@ -155,11 +155,11 @@ def write_images(self, batch_data, output_data, epoch):
 
                 y_img = Image.new("RGB", (200, 100))
                 draw = ImageDraw.Draw(y_img)
-                draw.text((10, 50), self.class_names[str(y_c)])
+                draw.text((10, 50), self.class_names.get(f"{y_c}", f"{y_c}"))
 
                 y_pred_img = Image.new("RGB", (200, 100), "green" if y_c == y_pred_c else "red")
                 draw = ImageDraw.Draw(y_pred_img)
-                draw.text((10, 50), self.class_names[str(y_pred_c)])
+                draw.text((10, 50), self.class_names.get(f"{y_pred_c}", f"{y_pred_c}"))
 
                 label_pred = [np.moveaxis(np.array(y_img), -1, 0), np.moveaxis(np.array(y_pred_img), -1, 0)]
                 img_tensor = make_grid(

setup.cfg

@@ -66,7 +66,7 @@ install_requires =
     python-jose[cryptography]==3.3.0
     bcrypt==3.2.2
     shapely==1.8.2
-    sklearn
+    scikit-learn
 
 [flake8]
 select = B,C,E,F,N,P,T4,W,B9