com.expleague.ml.methods.greedyMergeOptimization.RegularizedLossComparator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmll

Various ML methods implemented by myself and my students

There is a newer version: 1.4.9

package com.expleague.ml.methods.greedyMergeOptimization;

import java.util.Comparator;

/**
 * Created by noxoomo on 30/11/14.
 */
public class RegularizedLossComparator> implements ModelComparators {
  Loss loss;

  public RegularizedLossComparator(final Loss loss) {
    this.loss = loss;
  }

  @Override
  public Comparator regularizationComparator() {
    return new Comparator() {
      @Override
      public int compare(final Model left, final Model right) {
        return Double.compare(loss.regularization(left), loss.regularization(right));
      }
    };
  }

  @Override
  public Comparator targetComparator() {
    return new Comparator() {
      @Override
      public int compare(final Model left, final Model right) {
        return Double.compare(loss.target(left), loss.target(right));
      }
    };
  }

  @Override
  public Comparator scoreComparator() {
    return new Comparator() {
      @Override
      public int compare(final Model left, final Model right) {
        return Double.compare(loss.score(left), loss.score(right));
      }
    };
  }
}