com.expleague.ml.loss.multiclass.MCMicroF1Score Maven / Gradle / Ivy
package com.expleague.ml.loss.multiclass;
import com.expleague.commons.seq.IntSeq;
import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.math.Func;
import com.expleague.ml.data.set.DataSet;
/**
* User: qdeee
* Date: 09.04.14
*/
public class MCMicroF1Score extends Func.Stub implements ClassicMulticlassLoss {
private final MCMicroPrecision precision;
private final MCMicroRecall recall;
public MCMicroF1Score(final IntSeq target, final DataSet> owner) {
precision = new MCMicroPrecision(target, owner);
recall = new MCMicroRecall(target, owner);
}
public MCMicroF1Score(final Vec target, final DataSet> owner) {
precision = new MCMicroPrecision(target, owner);
recall = new MCMicroRecall(target, owner);
}
@Override
public double value(final Vec x) {
final double p = precision.value(x);
final double r = recall.value(x);
return (p + r) > 0 ? 2 * p * r / (p + r) : 0.;
}
@Override
public int dim() {
return precision.dim();
}
@Override
public DataSet> owner() {
return precision.owner();
}
@Override
public IntSeq labels() {
return precision.labels();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy