org.thales.punch.pl.plugins.udf.functions

Class KmeansSqdist

java.lang.Object

org.thales.punch.pl.plugins.udf.functions.KmeansSqdist

All Implemented Interfaces:

Serializable, org.apache.spark.sql.api.java.UDF4<scala.collection.mutable.WrappedArray<Double>,Integer,String,Integer,Double>

public class KmeansSqdist
extends Object
implements org.apache.spark.sql.api.java.UDF4<scala.collection.mutable.WrappedArray<Double>,Integer,String,Integer,Double>

This class is intended to be used in a Spark Sql Query statement as a function.
The function name is defined during UDFs registration: org.thales.punch.pml.plugins.sql.UdfRegistration and it should return a DataTypes.DoubleType

USAGE:

        Take as input 4 arguments:
                 WrappedArray<Double>:  features;
                Integer:  prediction;
                String:  modelName  (loaded inside settings -> spark.files);
                Integer:  stageNumber
        and return a Scala Double object
 

Author:

jonathan

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
KmeansSqdist()

Method Summary

Modifier and Type	Method and Description
Double	call(scala.collection.mutable.WrappedArray<Double> features, Integer prediction, String modelPath, Integer stageNumber)

Methods inherited from class java.lang.Object

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

Constructor Detail

KmeansSqdist

public KmeansSqdist()

Method Detail

call

public Double call(scala.collection.mutable.WrappedArray<Double> features,
                   Integer prediction,
                   String modelPath,
                   Integer stageNumber)
            throws Exception

Specified by:

call in interface org.apache.spark.sql.api.java.UDF4<scala.collection.mutable.WrappedArray<Double>,Integer,String,Integer,Double>

Throws:

Exception