org.jpmml.sparkml.testing

Class SparkMLEncoderBatchTest

java.lang.Object

org.jpmml.evaluator.testing.BatchTest

org.jpmml.evaluator.testing.ArchiveBatchTest

org.jpmml.converter.testing.ModelEncoderBatchTest

org.jpmml.sparkml.testing.SparkMLEncoderBatchTest

public class SparkMLEncoderBatchTest
extends org.jpmml.converter.testing.ModelEncoderBatchTest

Constructor Summary

Constructors

Constructor and Description
SparkMLEncoderBatchTest()
SparkMLEncoderBatchTest(com.google.common.base.Equivalence<Object> equivalence)

Method Summary

Modifier and Type	Method and Description
SparkMLEncoderBatch	createBatch(String algorithm, String dataset, Predicate<org.jpmml.evaluator.ResultField> columnFilter, com.google.common.base.Equivalence<Object> equivalence)
static void	createSparkSession()
static void	destroySparkSession()
static Predicate<org.jpmml.evaluator.ResultField>	excludePredictionFields()
static Predicate<org.jpmml.evaluator.ResultField>	excludePredictionFields(String predictionCol)
org.apache.spark.sql.SparkSession	getSparkSession()

Methods inherited from class org.jpmml.converter.testing.ModelEncoderBatchTest

evaluate, evaluate

Methods inherited from class org.jpmml.evaluator.testing.ArchiveBatchTest

evaluate, evaluate, evaluate, evaluate, getEquivalence

Methods inherited from class org.jpmml.evaluator.testing.BatchTest

excludeFields, excludeFields, log

Methods inherited from class java.lang.Object

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

Constructor Detail

SparkMLEncoderBatchTest

public SparkMLEncoderBatchTest()

SparkMLEncoderBatchTest

public SparkMLEncoderBatchTest(com.google.common.base.Equivalence<Object> equivalence)

Method Detail

getSparkSession

public org.apache.spark.sql.SparkSession getSparkSession()

See Also:

createSparkSession(), destroySparkSession()

createBatch

public SparkMLEncoderBatch createBatch(String algorithm,
                                       String dataset,
                                       Predicate<org.jpmml.evaluator.ResultField> columnFilter,
                                       com.google.common.base.Equivalence<Object> equivalence)

Specified by:

createBatch in class org.jpmml.evaluator.testing.ArchiveBatchTest

createSparkSession

public static void createSparkSession()

destroySparkSession

public static void destroySparkSession()

excludePredictionFields

public static Predicate<org.jpmml.evaluator.ResultField> excludePredictionFields()

excludePredictionFields

public static Predicate<org.jpmml.evaluator.ResultField> excludePredictionFields(String predictionCol)