Update test.py

Author: NoviceMAn-prog

test.py

@@ -28,6 +28,7 @@
 from tqdm import tqdm_notebook as tqdm
 import cv2
 from sklearn.utils import shuffle
+from sklearn.metrics import confusion_matrix
 from tqdm import tqdm
 import tifffile as tif
 from model import msrf