org.apache.ignite.ml.pipeline

Class PipelineMdl<K,V>

java.lang.Object

org.apache.ignite.ml.pipeline.PipelineMdl<K,V>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

All Implemented Interfaces:

Serializable, Function<Vector,Double>, IgniteFunction<Vector,Double>, Model<Vector,Double>

public class PipelineMdl<K,V>
extends Object
implements Model<Vector,Double>

Wraps the model produced by Pipeline.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
PipelineMdl()

Method Summary

Modifier and Type	Method and Description
Double	apply(Vector vector)
IgniteBiFunction<K,V,Vector>	getFeatureExtractor()
Model<Vector,Double>	getInternalMdl()
IgniteBiFunction<K,V,Double>	getLabelExtractor()
String	toString()
PipelineMdl<K,V>	withFeatureExtractor(IgniteBiFunction featureExtractor)
PipelineMdl<K,V>	withInternalMdl(Model<Vector,Double> internalMdl)
PipelineMdl<K,V>	withLabelExtractor(IgniteBiFunction<K,V,Double> lbExtractor)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.ignite.ml.Model

combine, toString

Methods inherited from interface java.util.function.Function

andThen, compose, identity

Constructor Detail

PipelineMdl

public PipelineMdl()

Method Detail

apply

public Double apply(Vector vector)

Specified by:

apply in interface Function<Vector,Double>

getFeatureExtractor

public IgniteBiFunction<K,V,Vector> getFeatureExtractor()

getLabelExtractor

public IgniteBiFunction<K,V,Double> getLabelExtractor()

getInternalMdl

public Model<Vector,Double> getInternalMdl()

withInternalMdl

public PipelineMdl<K,V> withInternalMdl(Model<Vector,Double> internalMdl)

withFeatureExtractor

public PipelineMdl<K,V> withFeatureExtractor(IgniteBiFunction featureExtractor)

withLabelExtractor

public PipelineMdl<K,V> withLabelExtractor(IgniteBiFunction<K,V,Double> lbExtractor)

toString

public String toString()

Overrides:

toString in class Object