/*
* Ivory: A Hadoop toolkit for web-scale information retrieval
*
* Licensed under the Apache License, Version 2.0 (the "License"); you
* may not use this file except in compliance with the License. You may
* obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package ivory.ltr;
import ivory.core.ConfigurationException;
import java.io.IOException;
import java.util.List;
/**
* @author Don Metzler
*
*/
public class Rank {
/**
* @param args
* @throws RankIRException
* @throws IOException
* @throws ClassNotFoundException
* @throws ConfigurationException
*/
public static void main(String[] args) throws IOException, ClassNotFoundException, ConfigurationException {
if(args.length != 2) {
System.out.println("Usage: Rank <test-file> <model-file>");
System.exit(-1);
}
String testFile = args[0];
String modelFile = args[1];
Instances instances = new Instances(testFile);
Model model = Model.readFromFile(modelFile);
List<Feature> features = model.getFeatures();
List<Double> weights = model.getWeights();
ScoreTable scores = new ScoreTable(instances);
for(int i = 0; i < features.size(); i++) {
scores = scores.translate(features.get(i), weights.get(i), 1.0);
}
System.out.print(scores);
}
}