net.librec.data.model

Class ArffDataModel

java.lang.Object

net.librec.conf.Configured

net.librec.data.model.AbstractDataModel

net.librec.data.model.ArffDataModel

All Implemented Interfaces:

Configurable, DataModel

public class ArffDataModel
extends AbstractDataModel
implements DataModel

A ArffDataModel represents a data access class to the ARFF format input.

Author:

WangYuFeng TangJiaxi and Ma Chen

Field Summary

Fields inherited from class net.librec.data.model.AbstractDataModel

context, dataAppender, dataConvertor, dataSplitter, LOG, testDataSet, trainDataSet, validDataSet

Fields inherited from class net.librec.conf.Configured

conf, CONF_DATA_COLUMN_FORMAT, CONF_DATA_INPUT_PATH, CONF_DFS_DATA_DIR

Constructor Summary

Constructors

Constructor and Description
ArffDataModel() Empty constructor.
ArffDataModel(Configuration conf) Initializes a newly created ArffDataModel object with configuration.

Method Summary

Methods

Modifier and Type	Method and Description
void	buildConvert() Build model.
protected void	buildSplitter() Build Splitter.
java.util.ArrayList<com.google.common.collect.BiMap<java.lang.String,java.lang.Integer>>	getAllFeaturesMappingData() Get all features mapping data.
DataSet	getDatetimeDataSet() Get datetime data set.
com.google.common.collect.BiMap<java.lang.String,java.lang.Integer>	getItemMappingData() Get item mapping data.
com.google.common.collect.BiMap<java.lang.String,java.lang.Integer>	getUserMappingData() Get user mapping data.

Methods inherited from class net.librec.data.model.AbstractDataModel

buildDataModel, buildFeature, getContext, getDataAppender, getDataSplitter, getTestDataSet, getTrainDataSet, getValidDataSet, loadDataModel, saveDataModel

Methods inherited from class net.librec.conf.Configured

getConf, setConf

Methods inherited from class java.lang.Object

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

Methods inherited from interface net.librec.data.DataModel

buildDataModel, getContext, getDataAppender, getDataSplitter, getTestDataSet, getTrainDataSet, getValidDataSet, loadDataModel, saveDataModel

Constructor Detail

ArffDataModel

public ArffDataModel()

Empty constructor.

ArffDataModel

public ArffDataModel(Configuration conf)

Initializes a newly created ArffDataModel object with configuration.

Parameters:

conf - the configuration for the model.

Method Detail

buildConvert

public void buildConvert()
                  throws LibrecException

Build model.

Specified by:

buildConvert in class AbstractDataModel

Throws:

LibrecException - if error occurs during building

buildSplitter

protected void buildSplitter()
                      throws LibrecException

Build Splitter.

Overrides:

buildSplitter in class AbstractDataModel

Throws:

LibrecException - if error occurs when building splitter.

getUserMappingData

public com.google.common.collect.BiMap<java.lang.String,java.lang.Integer> getUserMappingData()

Get user mapping data.

Specified by:

getUserMappingData in interface DataModel

Returns:

the user {raw id, inner id} map of data model.

getItemMappingData

public com.google.common.collect.BiMap<java.lang.String,java.lang.Integer> getItemMappingData()

Get item mapping data.

Specified by:

getItemMappingData in interface DataModel

Returns:

the item {raw id, inner id} map of data model.

getDatetimeDataSet

public DataSet getDatetimeDataSet()

Description copied from interface: DataModel

Get datetime data set.

Specified by:

getDatetimeDataSet in interface DataModel

Returns:

the datetime data set of data model.

getAllFeaturesMappingData

public java.util.ArrayList<com.google.common.collect.BiMap<java.lang.String,java.lang.Integer>> getAllFeaturesMappingData()

Get all features mapping data.

Returns:

the item {raw id, inner id} map of data model.