Skip to content

Commit 1f1f741

Browse files
committed
Fixed broken labeling
1 parent 4084869 commit 1f1f741

File tree

1 file changed

+10
-1
lines changed
  • src/main/java/ml/anon/model/anonymization

1 file changed

+10
-1
lines changed

src/main/java/ml/anon/model/anonymization/Label.java

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,16 @@
77
* Created by mirco on 20.05.17.
88
*/
99
public enum Label {
10-
PERSON, LOCATION, ORGANIZATION, MISC, LICENCE_PLATE, E_MAIL, TELEPHONE_NUMBER, URL, IP, BIRTHDATE, IBAN, CLIENT_NUMBER;
10+
PERSON, LOCATION, ORGANIZATION, MISC, LICENCE_PLATE, E_MAIL, TELEPHONE_NUMBER, URL, IP, BIRTHDATE, IBAN, CLIENT_NUMBER, UNKNOWN;
11+
12+
public static Label getOrDefault(String label, Label def) {
13+
try {
14+
return Label.valueOf(label);
15+
} catch (IllegalArgumentException e) {
16+
e.printStackTrace();
17+
return def;
18+
}
19+
}
1120

1221
public static List<Label> getAll() {
1322
return Arrays.asList(Label.values());

0 commit comments

Comments
 (0)