org.lenskit.eval.traintest.predict

Class RMSEPredictMetric

java.lang.Object

org.lenskit.eval.traintest.metrics.Metric<X>

org.lenskit.eval.traintest.predict.PredictMetric<RMSEPredictMetric.Context>

org.lenskit.eval.traintest.predict.RMSEPredictMetric

public class RMSEPredictMetric
extends PredictMetric<RMSEPredictMetric.Context>

Evaluate a recommender’s prediction accuracy with RMSE.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
class	RMSEPredictMetric.Context

Constructor Summary

Constructors

Constructor and Description
RMSEPredictMetric()

Method Summary

Modifier and Type	Method and Description
RMSEPredictMetric.Context	createContext(AlgorithmInstance algorithm, DataSet dataSet, Recommender recommender) Create the context for an experimental condition (algorithm/data set pair).
MetricResult	getAggregateMeasurements(RMSEPredictMetric.Context context) Get the aggregate results from an accumulator.
MetricResult	measureUser(TestUser user, ResultMap predictions, RMSEPredictMetric.Context context) Measure a single result.

Methods inherited from class org.lenskit.eval.traintest.metrics.Metric

getAggregateColumnLabels, getColumnLabels, getRequiredRoots

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

@Nullable
public RMSEPredictMetric.Context createContext(AlgorithmInstance algorithm,
                                                          DataSet dataSet,
                                                          Recommender recommender)

Description copied from class: Metric

Create the context for an experimental condition (algorithm/data set pair). The default implementation returns null.

Overrides:

createContext in class Metric<RMSEPredictMetric.Context>

Parameters:

algorithm - The algorithm.

dataSet - The data set.

recommender - 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 the individual measurement methods. If the metric does need to accumulate any results, this method can return null.

measureUser

@Nonnull
public MetricResult measureUser(TestUser user,
                                          ResultMap predictions,
                                          RMSEPredictMetric.Context context)

Description copied from class: PredictMetric

Measure a single result. The result may come from either prediction or recommendation.

Specified by:

measureUser in class PredictMetric<RMSEPredictMetric.Context>

Parameters:

user - The user’s test data.

predictions - The predictions.

Returns:

A list of fields to add to the result’s output.

getAggregateMeasurements

@Nonnull
public MetricResult getAggregateMeasurements(RMSEPredictMetric.Context context)

Description copied from class: Metric

Get the aggregate results from an accumulator. The default implementation returns MetricResult.empty().

Overrides:

getAggregateMeasurements in class Metric<RMSEPredictMetric.Context>

Parameters:

context - The context for an experimental condition.

Returns:

The aggregate results from the accumulator.