-
- Downloads
[SPARK-21729][ML][TEST] Generic test for ProbabilisticClassifier to ensure...
[SPARK-21729][ML][TEST] Generic test for ProbabilisticClassifier to ensure consistent output columns ## What changes were proposed in this pull request? Add test for prediction using the model with all combinations of output columns turned on/off. Make sure the output column values match, presumably by comparing vs. the case with all 3 output columns turned on. ## How was this patch tested? Test updated. Author: WeichenXu <weichen.xu@databricks.com> Author: WeichenXu <WeichenXu123@outlook.com> Closes #19065 from WeichenXu123/generic_test_for_prob_classifier.
Showing
- mllib/src/test/scala/org/apache/spark/ml/classification/DecisionTreeClassifierSuite.scala 3 additions, 0 deletions...spark/ml/classification/DecisionTreeClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/GBTClassifierSuite.scala 3 additions, 0 deletions...g/apache/spark/ml/classification/GBTClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala 6 additions, 0 deletions...che/spark/ml/classification/LogisticRegressionSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifierSuite.scala 2 additions, 0 deletions.../classification/MultilayerPerceptronClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/NaiveBayesSuite.scala 6 additions, 0 deletions.../org/apache/spark/ml/classification/NaiveBayesSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/ProbabilisticClassifierSuite.scala 60 additions, 0 deletions...park/ml/classification/ProbabilisticClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/RandomForestClassifierSuite.scala 2 additions, 0 deletions...spark/ml/classification/RandomForestClassifierSuite.scala
Loading
Please register or sign in to comment