Examples with MLPriorProbClassifier edu.neu.ccs.pyramid.multilabel_classification.MLPriorProbClassifier used on opensource projects

Example 1 with MLPriorProbClassifier

use of edu.neu.ccs.pyramid.multilabel_classification.MLPriorProbClassifier in project pyramid by cheng-li.

the class HMLGBTrainer method setPriorProbs.

/**
     * not sure whether this is good for performance
     * start with prior probabilities
     * should be called before setTrainConfig
     */
private void setPriorProbs(MultiLabelClfDataSet dataSet) {
    MLPriorProbClassifier priorProbClassifier = new MLPriorProbClassifier(dataSet.getNumClasses());
    priorProbClassifier.fit(dataSet);
    double[] probs = priorProbClassifier.getClassProbs();
    this.setPriorProbs(probs);
}

Example 2 with MLPriorProbClassifier

use of edu.neu.ccs.pyramid.multilabel_classification.MLPriorProbClassifier in project pyramid by cheng-li.

the class IMLGBTrainer method setPriorProbs.

/**
     * not sure whether this is good for performance
     * start with prior probabilities
     * should be called before setTrainConfig
     */
private void setPriorProbs(MultiLabelClfDataSet dataSet) {
    MLPriorProbClassifier priorProbClassifier = new MLPriorProbClassifier(dataSet.getNumClasses());
    priorProbClassifier.fit(dataSet);
    double[] probs = priorProbClassifier.getClassProbs();
    this.setPriorProbs(probs);
}

Example 3 with MLPriorProbClassifier

use of edu.neu.ccs.pyramid.multilabel_classification.MLPriorProbClassifier in project pyramid by cheng-li.

the class AdaBoostMHTrainer method setPriorProbs.

/**
     * not sure whether this is good for performance
     * start with prior probabilities
     * should be called before setTrainConfig
     */
private void setPriorProbs(MultiLabelClfDataSet dataSet) {
    MLPriorProbClassifier priorProbClassifier = new MLPriorProbClassifier(dataSet.getNumClasses());
    priorProbClassifier.fit(dataSet);
    double[] probs = priorProbClassifier.getClassProbs();
    this.setPriorProbs(probs);
}