error handling when input sample is not found in the dataset

ntucllab · Dec 31, 2015 · a798c97 · a798c97
1 parent b99dcf2
commit a798c97
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libact/labelers/ideal_labeler.py b/libact/labelers/ideal_labeler.py
@@ -26,4 +26,6 @@ def __init__(self, dataset, **kwargs):
         self.y = y
 
     def label(self, feature):
+        if np.count_nonzero([np.array_equal(x, feature) for x in self.X])==0
+            raise ValueError("No matching sample found in the dataset.")
         return self.y[np.where([np.array_equal(x, feature) for x in self.X])[0]]