edu.columbia.tjw.item.spark

Class ClassificationModelEvaluator

java.lang.Object

edu.columbia.tjw.item.spark.ClassificationModelEvaluator

public class ClassificationModelEvaluator
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ClassificationModelEvaluator.EntropyResult
static class	ClassificationModelEvaluator.EvaluationResult

Constructor Summary

Constructors

Constructor and Description
ClassificationModelEvaluator()

Method Summary

Modifier and Type	Method and Description
static <W extends org.apache.spark.ml.classification.ProbabilisticClassificationModel<org.apache.spark.ml.linalg.Vector,W>,M extends org.apache.spark.ml.classification.ProbabilisticClassifier<org.apache.spark.ml.linalg.Vector,M,W>> ClassificationModelEvaluator.EvaluationResult	evaluate(M classifier_, String label_, org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> fitting, org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> testing, long prngSeed_, int[] layers_)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ClassificationModelEvaluator

public ClassificationModelEvaluator()

Method Detail

evaluate

public static <W extends org.apache.spark.ml.classification.ProbabilisticClassificationModel<org.apache.spark.ml.linalg.Vector,W>,M extends org.apache.spark.ml.classification.ProbabilisticClassifier<org.apache.spark.ml.linalg.Vector,M,W>> ClassificationModelEvaluator.EvaluationResult evaluate(M classifier_,
                                                                                                                                                                                                                                                                                                      String label_,
                                                                                                                                                                                                                                                                                                      org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> fitting,
                                                                                                                                                                                                                                                                                                      org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> testing,
                                                                                                                                                                                                                                                                                                      long prngSeed_,
                                                                                                                                                                                                                                                                                                      int[] layers_)