com.shijingsh.ai.evaluate.rating.MSEEvaluator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of shijingsh-ai-evaluate Show documentation

The newest version!

package com.shijingsh.ai.evaluate.rating;

import java.util.Iterator;

import com.shijingsh.ai.evaluate.RatingEvaluator;

import com.shijingsh.ai.evaluate.RatingEvaluator;
import it.unimi.dsi.fastutil.floats.FloatList;

/**
 * 均方误差
 *
 *  * MSE = Mean Squared Error
 * 
 *
 * @author Birdy
 */
public class MSEEvaluator extends RatingEvaluator {

    public MSEEvaluator(float minimum, float maximum) {
        super(minimum, maximum);
    }

    @Override
    protected float measure(FloatList checkCollection, FloatList rateList) {
        float value = 0F;
        Iterator iterator = checkCollection.iterator();
        for (float estimate : rateList) {
            float score = iterator.next();
            if (estimate > maximum) {
                estimate = maximum;
            }
            if (estimate < minimum) {
                estimate = minimum;
            }
            value += Math.pow(score - estimate, 2);
        }
        return value;
    }

}