org.apache.ignite.ml.mleap

Class MLeapModel

java.lang.Object

org.apache.ignite.ml.mleap.MLeapModel

All Implemented Interfaces:

AutoCloseable, Model<NamedVector,Double>

public class MLeapModel
extends Object
implements Model<NamedVector,Double>

MLeap model imported and wrapped to be compatible with Apache Ignite infrastructure.

Constructor Summary

Constructors

Constructor and Description
MLeapModel(ml.combust.mleap.runtime.frame.Transformer transformer, List<String> schema, String outputFieldName) Constructs a new instance of MLeap model.

Method Summary

Modifier and Type	Method and Description
void	close()
double	predict(ml.combust.mleap.runtime.frame.DefaultLeapFrame inputFrame) Makes a prediction using MLeap API.
double	predict(Double[] input) Makes a prediction using default column order specified in schema.
Double	predict(NamedVector input)

Methods inherited from class java.lang.Object

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

Constructor Detail

MLeapModel

public MLeapModel(ml.combust.mleap.runtime.frame.Transformer transformer,
                  List<String> schema,
                  String outputFieldName)

Constructs a new instance of MLeap model.

Parameters:

transformer - MLpeap model (transformer in terms of MLeap).

schema - List of field names.

outputFieldName - Output field name.

Method Detail

predict

public Double predict(NamedVector input)

Specified by:

predict in interface Model<NamedVector,Double>

predict

public double predict(Double[] input)

Makes a prediction using default column order specified in schema.

Parameters:

input - Input arguments.

Returns:

Prediction result.

predict

public double predict(ml.combust.mleap.runtime.frame.DefaultLeapFrame inputFrame)

Makes a prediction using MLeap API.

Parameters:

inputFrame - Input MLeap frame.

Returns:

Prediction result.

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Model<NamedVector,Double>