Class LogisticScore

src/com/winvector/logistic/LogisticScore.java:38–232

import com.winvector.util.TrivialReader;

public class LogisticScore {
	private static final String DATAURIKEY = "dataURI";
	private static final String DATASEP = "sep";
	private static final String DATAHDLKEY = "dataHDL";
	private static final String DATATBLKEY = "dataTBL";
	private static final String MODELKEY = "modelFile";
	private static final String RESULTFILEKEY = "resultFile";

	private static CommandLine parseCommandLine(final String[] args) throws org.apache.commons.cli.ParseException {
		final CommandLineParser clparser = new GnuParser();
		final Options cloptions = new Options();
		cloptions.addOption(MODELKEY,true,"file to read serialized model from");
		cloptions.addOption(DATAURIKEY,true,"URI to get scoring data from");
		cloptions.addOption(DATASEP,true,"(optional) data input separator");
		cloptions.addOption(DATAHDLKEY,true,"XML file to get JDBC connection to scoring data table");
		cloptions.addOption(DATATBLKEY,true,"table to use from database for scoring data");
		cloptions.addOption(RESULTFILEKEY,true,"file to write results to");
		for(final String optkey: new String[] {MODELKEY, RESULTFILEKEY} ) {
			cloptions.getOption(optkey).setRequired(true);
		}
		final HelpFormatter hf = new HelpFormatter();
		final CommandLine cl = clparser.parse(cloptions, args);
		if((cl.getOptionValue(DATAURIKEY)==null)==(cl.getOptionValue(DATAHDLKEY)==null)) {
			hf.printHelp("com.winvector.logistic.LogisticScore", cloptions);
			throw new org.apache.commons.cli.ParseException("Must set exacty one of --" + DATAURIKEY + " or --" + DATAHDLKEY);
		}
		if((cl.getOptionValue(DATAHDLKEY)!=null)&&(cl.getOptionValue(DATATBLKEY)==null)) {
			hf.printHelp("com.winvector.logistic.LogisticScore", cloptions);
			throw new org.apache.commons.cli.ParseException("If --" + DATAHDLKEY + " is set then must specify a table with --" + DATATBLKEY);
		}
		return cl;
	}

	/**
	 * @param args
	 * @throws IOException 
	 * @throws FileNotFoundException 
	 * @throws ClassNotFoundException 
	 * @throws URISyntaxException 
	 * @throws ParseException 
	 * @throws SQLException 
	 */
	public static void main(String[] args) throws FileNotFoundException, IOException, ClassNotFoundException, URISyntaxException, ParseException, SQLException {
		final Log log = LogFactory.getLog(LogisticScore.class);
		log.info("start LogisticScore\t" + new Date());
		final CommandLine cl = parseCommandLine(args);
		final File modelFile = new File(cl.getOptionValue(MODELKEY));
		final File resultFile = new File(cl.getOptionValue(RESULTFILEKEY));
		log.info("cwd: " + new File(".").getAbsolutePath());
		log.info("reading model: " + modelFile.getAbsolutePath());
		final ObjectInputStream ois = new ObjectInputStream(new FileInputStream(modelFile));
		final Model model = (Model)ois.readObject();		
		ois.close();
		log.info("model:\n" + model.config.formatSoln(model.coefs));
		DBHandle handle = null;
		Statement stmt = null;
		final Iterable<BurstMap> testSource;
		if(cl.getOptionValue(DATAURIKEY)!=null) {