public static class SpamFilterFunction implements
org.apache.spark.api.java.function.Function<row7Struct, Boolean>{
private static final long serialVersionUID = 1L;
@Override
public Boolean call(row7Struct row7) throws Exception {
return row7.reallabel.equals("spam");
}
}
// 'negative': ham
// 'positive': spam
// 'false' means the real label & predicted label are different
// 'true' means the real label & predicted label are the same
public static class TrueNegativeFunction implements
org.apache.spark.api.java.function.Function<row7Struct, Boolean>{
private static final long serialVersionUID = 1L;
@Override
public Boolean call(row7Struct row7) throws Exception {
return (row7.label.equals("ham") && row7.reallabel.equals("ham"));
}
}
public static class TruePositiveFunction implements
org.apache.spark.api.java.function.Function<row7Struct, Boolean>{
private static final long serialVersionUID = 1L;
@Override
public Boolean call(row7Struct row7) throws Exception {
// true positive cases
return (row7.label.equals("spam") && row7.reallabel.equals("spam"));
}
}
public static class FalseNegativeFunction implements
org.apache.spark.api.java.function.Function<row7Struct, Boolean>{
private static final long serialVersionUID = 1L;
@Override
public Boolean call(row7Struct row7) throws Exception {
// false positive cases
return (row7.label.equals("spam") && row7.reallabel.equals("ham"));
}
}
public static class FalsePositiveFunction implements
org.apache.spark.api.java.function.Function<row7Struct, Boolean>{
private static final long serialVersionUID = 1L;
@Override
public Boolean call(row7Struct row7) throws Exception {
// false positive cases
return (row7.label.equals("ham") && row7.reallabel.equals("spam"));
}
}