ciir.umass.edu.learning.neuralnet.PropParameter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ranklib Show documentation

The newest version!

/*===============================================================================
 * Copyright (c) 2010-2012 University of Massachusetts.  All Rights Reserved.
 *
 * Use of the RankLib package is subject to the terms of the software license set
 * forth in the LICENSE file included with this software, and also available at
 * http://people.cs.umass.edu/~vdang/ranklib_license.html
 *===============================================================================
 */

package ciir.umass.edu.learning.neuralnet;

public class PropParameter {
    //RankNet
    protected int current = -1;//index of current data point in the ranked list
    protected int[][] pairMap = null;

    public PropParameter(final int current, final int[][] pairMap) {
        this.current = current;
        this.pairMap = pairMap;
    }

    //LambdaRank: RankNet + the following
    protected float[][] pairWeight = null;
    protected float[][] targetValue = null;

    public PropParameter(final int current, final int[][] pairMap, final float[][] pairWeight, final float[][] targetValue) {
        this.current = current;
        this.pairMap = pairMap;
        this.pairWeight = pairWeight;
        this.targetValue = targetValue;
    }

    //ListNet
    protected float[] labels = null;//relevance label

    public PropParameter(final float[] labels) {
        this.labels = labels;
    }
}