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) Make a prediction for the specified input arguments.

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)

Make a prediction for the specified input arguments.

Specified by:

predict in interface Model<NamedVector,Double>

Parameters:

input - Input arguments.

Returns:

Prediction result.

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>