org.terrier.matching

Class LinearModelMatching

java.lang.Object

org.terrier.matching.LearnedModelMatching

org.terrier.matching.LinearModelMatching

All Implemented Interfaces:

Matching

public class LinearModelMatching
extends LearnedModelMatching

Applies a linear learned model to a FeaturedResultSet. Learned model files take the format featureid:weight

Since:

4.0

Author:

Craig Macdonald

Field Summary

Fields inherited from class org.terrier.matching.LearnedModelMatching

increment, logger, parent, score_is_feature

Constructor Summary

Constructors

Constructor and Description
LinearModelMatching(Index _index, Matching _parent)
LinearModelMatching(Index _index, Matching _parent, double[] _weights)
LinearModelMatching(Index _index, Matching _parent, String _modelFilename)

Method Summary

Methods

Modifier and Type	Method and Description
protected void	applyModel(int N, double[] in_scores, int F, double[][] features, double[] out_scores) Apply the loaded learned model for identifying the top N documents.
static double[]	loadFeatureWeights(String input)

Methods inherited from class org.terrier.matching.LearnedModelMatching

getInfo, match, setCollectionStatistics

Methods inherited from class java.lang.Object

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

Constructor Detail

LinearModelMatching

public LinearModelMatching(Index _index,
                   Matching _parent,
                   double[] _weights)
                    throws Exception

Throws:

Exception

LinearModelMatching

public LinearModelMatching(Index _index,
                   Matching _parent,
                   String _modelFilename)
                    throws Exception

Throws:

Exception

LinearModelMatching

public LinearModelMatching(Index _index,
                   Matching _parent)
                    throws Exception

Throws:

Exception

Method Detail

applyModel

protected void applyModel(int N,
              double[] in_scores,
              int F,
              double[][] features,
              double[] out_scores)

Description copied from class: LearnedModelMatching

Apply the loaded learned model for identifying the top N documents.

Specified by:

applyModel in class LearnedModelMatching

Parameters:

N - - number of documents

in_scores - - sample scores

F - number of features

features - indexed by feature THEN document

out_scores - - final scores

loadFeatureWeights

public static double[] loadFeatureWeights(String input)
                                   throws IOException

Throws:

IOException