com.expleague.ml.cli.modes.impl.EvaluateModel 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.cli.modes.impl;

import com.expleague.commons.math.vectors.Mx;
import com.expleague.ml.cli.builders.data.impl.DataBuilderClassic;
import com.expleague.commons.io.StreamTools;
import com.expleague.commons.math.Trans;
import com.expleague.ml.TargetFunc;
import com.expleague.ml.cli.builders.methods.grid.GridBuilder;
import com.expleague.ml.cli.modes.AbstractMode;
import com.expleague.ml.cli.modes.CliPoolReaderHelper;
import com.expleague.ml.data.tools.DataTools;
import com.expleague.ml.data.tools.Pool;
import com.expleague.ml.BFGrid;
import com.expleague.ml.io.ModelsSerializationRepository;
import com.expleague.ml.meta.DSItem;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.MissingArgumentException;

import java.io.File;
import java.io.IOException;

import static com.expleague.ml.cli.JMLLCLI.*;

/**
 * User: qdeee
 * Date: 16.09.15
 */
public class EvaluateModel extends AbstractMode {

  public void run(final CommandLine command) throws MissingArgumentException, IOException {
    if (!command.hasOption(MODEL_OPTION))
      throw new MissingArgumentException("Please provide 'MODEL_OPTION'");
    if (!command.hasOption(METRICS_OPTION))
      throw new MissingArgumentException("Please provide 'METRICS_OPTION'");
    if (!command.hasOption(TEST_OPTION))
      throw new MissingArgumentException("Please provide 'TEST_OPTION'");

    final Trans model;
    { // loading model
      final ModelsSerializationRepository serializationRepository;
      if (command.hasOption(GRID_OPTION)) {
        final GridBuilder gridBuilder = new GridBuilder();
        gridBuilder.setGrid(BFGrid.CONVERTER.convertFrom(StreamTools.readFile(new File(command.getOptionValue(GRID_OPTION)))));
        serializationRepository = new ModelsSerializationRepository(gridBuilder.create());
      }
      else {
        serializationRepository = new ModelsSerializationRepository();
      }
      try {
        model = DataTools.readModel(command.getOptionValue(MODEL_OPTION), serializationRepository);
      }
      catch (ClassNotFoundException e) {
        throw new RuntimeException(e);
      }
    }
    final Pool pool;
    { // loading data
      final DataBuilderClassic dataBuilder = new DataBuilderClassic();
      dataBuilder.setLearnPath(command.getOptionValue(TEST_OPTION));
      CliPoolReaderHelper.setPoolReader(command, dataBuilder);
      //noinspection unchecked
      pool = dataBuilder.create().getFirst();
    }
    {
      final Mx mx = model.transAll(pool.vecData().data());
      for (final String metricName : command.getOptionValues(METRICS_OPTION)) {
        final TargetFunc target = pool.targetByName(metricName);
        target.printResult(mx, System.out);
      }
    }
  }
}