com.expleague.ml.cli.modes.impl.PrintPoolInfo Maven / Gradle / Ivy

Go to download
package com.expleague.ml.cli.modes.impl;

import com.expleague.ml.cli.builders.data.DataBuilder;
import com.expleague.ml.cli.builders.data.impl.DataBuilderClassic;
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 org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.MissingArgumentException;

import java.io.IOException;

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


/**
 * User: qdeee
 * Date: 16.09.15
 */
public class PrintPoolInfo extends AbstractMode {
  public void run(final CommandLine command) throws MissingArgumentException, IOException {
    if (!command.hasOption(LEARN_OPTION)) {
      throw new MissingArgumentException("Please provide: learn_option");
    }

    final DataBuilder builder = new DataBuilderClassic();
    builder.setLearnPath(command.getOptionValue(LEARN_OPTION));
    CliPoolReaderHelper.setPoolReader(command, builder);
    final Pool pool = builder.create().getFirst();
    System.out.println(DataTools.getPoolInfo(pool));
  }
}