org.lenskit.eval.traintest

Class DataSet

java.lang.Object

org.lenskit.eval.traintest.DataSet

public class DataSet
extends Object

A train-test data set.

Constructor Summary

Constructors

Constructor and Description
DataSet(String name, StaticDataSource train, StaticDataSource query, StaticDataSource test, UUID grp, Map<String,Object> attrs) Create a new data set.

Method Summary

Modifier and Type	Method and Description
void	configure(LenskitConfiguration config) Configure LensKit to have the training data from this data source.
DataSetBuilder	copyBuilder() Create a new builder initialized with this data set’s values.
static DataSetBuilder	copyBuilder(DataSet data) Create a new builder initialized with this data set’s values.
static List<DataSet>	fromJSON(com.fasterxml.jackson.databind.JsonNode json, URI base) Load one or more data sets from JSON data.
LongSet	getAllItems()
Map<String,Object>	getAttributes() Get the data set attributes (used for identification in output).
UUID	getIsolationGroup() Get the isolation group ID for this data set.
String	getName() Get the data set name.
StaticDataSource	getQueryData() Get the query data.
StaticDataSource	getTestData() Get the training data.
StaticDataSource	getTrainingData() Get the training data.
static List<DataSet>	load(Path file) Load one or more data sets from a YAML manifest file.
static List<DataSet>	load(URL url) Load one or more data sets from a YAML manifest file.
static DataSetBuilder	newBuilder() Create a new generic train-test data set builder.
static DataSetBuilder	newBuilder(String name) Create a new generic train-test data set builder.
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

DataSet

public DataSet(@Nonnull
               String name,
               @Nonnull
               StaticDataSource train,
               @Nullable
               StaticDataSource query,
               @Nonnull
               StaticDataSource test,
               @Nonnull
               UUID grp,
               Map<String,Object> attrs)

Create a new data set.

Parameters:

name - The name.

train - The training source.

query - The query source (if any).

test - The test data source.

grp - The data set isolation group.

attrs - The data set attributes.

Method Detail

getName

@Nonnull
public String getName()

Get the data set name.

Returns:

A name for the data set. Used in the output file.

getAttributes

public Map<String,Object> getAttributes()

Get the data set attributes (used for identification in output).

Returns:

A key → value map of the attributes used to identify this data set. For example, a crossfold data set may include the source name and fold number.

getIsolationGroup

public UUID getIsolationGroup()

Get the isolation group ID for this data set. Data sets in the same group will be allowed to run in parallel. This is used to implement data set isolation.

Returns:

The group ID for this data set.

getTestData

@Nonnull
public StaticDataSource getTestData()

Get the training data.

Returns:

A data source containing the test data.

getTrainingData

@Nonnull
public StaticDataSource getTrainingData()

Get the training data.

Returns:

A data source containing the training data.

getQueryData

@Nullable
public StaticDataSource getQueryData()

Get the query data.

Returns:

A data source containing the query data.

getAllItems

public LongSet getAllItems()

configure

public void configure(LenskitConfiguration config)

Configure LensKit to have the training data from this data source.

Parameters:

config - A configuration in which the training data for this data set should be configured.

toString

public String toString()

Overrides:

toString in class Object

newBuilder

public static DataSetBuilder newBuilder()

Create a new generic train-test data set builder.

Returns:

The new builder.

newBuilder

public static DataSetBuilder newBuilder(String name)

Create a new generic train-test data set builder.

Parameters:

name - The data set name.

Returns:

The new builder.

copyBuilder

public DataSetBuilder copyBuilder()

Create a new builder initialized with this data set’s values.

Returns:

A new builder initialized to make a copy of this data set definition.

copyBuilder

public static DataSetBuilder copyBuilder(DataSet data)

Create a new builder initialized with this data set’s values.

Returns:

A new builder initialized to make a copy of this data set definition.

fromJSON

public static List<DataSet> fromJSON(com.fasterxml.jackson.databind.JsonNode json,
                                     URI base)
                              throws IOException

Load one or more data sets from JSON data.

Throws:

IOException

load

public static List<DataSet> load(Path file)
                          throws IOException

Load one or more data sets from a YAML manifest file.

Parameters:

file - The path to the YAML manifest file.

Returns:

The list of data sets.

Throws:

IOException

load

public static List<DataSet> load(URL url)
                          throws IOException

Load one or more data sets from a YAML manifest file.

Parameters:

url - The URL of a the YAML manifest file.

Returns:

The list of data sets.

Throws:

IOException