From a798c97b8cd64ff60141f55773aef6ea285c9dc7 Mon Sep 17 00:00:00 2001 From: tungen Date: Thu, 31 Dec 2015 16:41:20 +0800 Subject: [PATCH] error handling when input sample is not found in the dataset --- libact/labelers/ideal_labeler.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/libact/labelers/ideal_labeler.py b/libact/labelers/ideal_labeler.py index dc89efc..a45d0b3 100644 --- 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]]