org.grouplens.lenskit.eval.metrics.predict

Class RMSEPredictMetric

java.lang.Object

org.grouplens.lenskit.eval.metrics.AbstractMetric<RMSEPredictMetric.Context,RMSEPredictMetric.AggregateResult,RMSEPredictMetric.UserResult>

org.grouplens.lenskit.eval.metrics.predict.RMSEPredictMetric

All Implemented Interfaces:

Closeable, AutoCloseable, Metric<RMSEPredictMetric.Context>

public class RMSEPredictMetric
extends AbstractMetric<RMSEPredictMetric.Context,RMSEPredictMetric.AggregateResult,RMSEPredictMetric.UserResult>

Evaluate a recommender's prediction accuracy with RMSE.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	RMSEPredictMetric.AggregateResult
class	RMSEPredictMetric.Context
static class	RMSEPredictMetric.UserResult

Constructor Summary

Constructors

Constructor and Description
RMSEPredictMetric()

Method Summary

Modifier and Type	Method and Description
RMSEPredictMetric.Context	createContext(Attributed algo, TTDataSet ds, Recommender rec) Create the context for an experimental condition (algorithm/data set pair).
RMSEPredictMetric.UserResult	doMeasureUser(TestUser user, RMSEPredictMetric.Context context) Measure a user with typed results.
protected RMSEPredictMetric.AggregateResult	getTypedResults(RMSEPredictMetric.Context context) Get the typed results from an accumulator.

Methods inherited from class org.grouplens.lenskit.eval.metrics.AbstractMetric

close, getColumnLabels, getPrefix, getResults, getSuffix, getUserColumnLabels, measureUser

Methods inherited from class java.lang.Object

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

Constructor Detail

RMSEPredictMetric

public RMSEPredictMetric()

Method Detail

createContext

public RMSEPredictMetric.Context createContext(Attributed algo,
                                               TTDataSet ds,
                                               Recommender rec)

Description copied from interface: Metric

Create the context for an experimental condition (algorithm/data set pair).

Parameters:

algo - The algorithm.

ds - The data set.

rec - The LensKit recommender, if applicable. This can be null for an external algorithm that does not provide a LensKit recommender.

Returns:

The accumulator. This will be passed to Metric.measureUser(TestUser, Object). If the metric does not accumulate any results, this method can return null.

doMeasureUser

public RMSEPredictMetric.UserResult doMeasureUser(TestUser user,
                                                  RMSEPredictMetric.Context context)

Description copied from class: AbstractMetric

Measure a user with typed results.

Specified by:

doMeasureUser in class AbstractMetric<RMSEPredictMetric.Context,RMSEPredictMetric.AggregateResult,RMSEPredictMetric.UserResult>

Parameters:

user - The user to measure.

context - The context.

Returns:

The results of measuring the user, or null to emit NAs for the user.

getTypedResults

protected RMSEPredictMetric.AggregateResult getTypedResults(RMSEPredictMetric.Context context)

Description copied from class: AbstractMetric

Get the typed results from an accumulator.

Specified by:

getTypedResults in class AbstractMetric<RMSEPredictMetric.Context,RMSEPredictMetric.AggregateResult,RMSEPredictMetric.UserResult>

Parameters:

context - The context.

Returns:

The results accumulated for this experiment, or null to emit NAs.