diff --git a/src/main/java/com/digitalpebble/classification/Lexicon.java b/src/main/java/com/digitalpebble/classification/Lexicon.java index 3a59a75..d608265 100755 --- a/src/main/java/com/digitalpebble/classification/Lexicon.java +++ b/src/main/java/com/digitalpebble/classification/Lexicon.java @@ -303,7 +303,11 @@ private void loadFromFile(String filename) throws IOException { .readLine()); this.normalizeVector = Boolean.parseBoolean(reader.readLine()); this.classifierType = reader.readLine(); - this.labels = Arrays.asList(reader.readLine().split(" ")); + this.labels = new ArrayList(); + // Need -1 to handle the case where the last label is zero length + this.labels.addAll(Arrays.asList(reader.readLine().split(" ", -1))); + // Remove the extra entry created by the terminating space + this.labels.remove(labels.size() - 1); String[] tmp = reader.readLine().split(" "); for (String f : tmp) { // see if there is a custom weight for it