Skip to content

Commit 5412a2d

Browse files
committed
fix to allow missing values
1 parent dc4f8ee commit 5412a2d

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

JSAT/src/jsat/classifiers/ClassificationDataSet.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -337,8 +337,11 @@ public void addDataPoint(DataPoint dp, int classification)
337337
throw new RuntimeException("Data point does not contain enough categorical data points");
338338

339339
for(int i = 0; i < dp.getCategoricalValues().length; i++)
340-
if(!categories[i].isValidCategory(dp.getCategoricalValues()[i]))
340+
{
341+
int val = dp.getCategoricalValues()[i];
342+
if(!categories[i].isValidCategory(val) && val >= 0)
341343
throw new RuntimeException("Categoriy value given is invalid");
344+
}
342345

343346
datapoints.add(dp);
344347
category.add(classification);

0 commit comments

Comments
 (0)