Fix misc.py import for confusion matrix plot

kr-colab · Nov 13, 2023 · 3a968c3 · 3a968c3
1 parent a9461a8
commit 3a968c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diploshic/diploSHIC b/diploshic/diploSHIC
@@ -534,7 +534,7 @@ if argsDict["mode"] == "train":
     print("diploSHIC loss: %f" % score[0])
     print("diploSHIC accuracy: %f" % score[1])
     if confusionFile:
-        from misc import plot_confusion_matrix
+        from diploshic.misc import plot_confusion_matrix
         import matplotlib.pyplot as plt
 
         plot_confusion_matrix(