cc.mallet.classify.tui.Classifier2Info Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jcore-mallet-2.0.9 Show documentation

MALLET is a Java-based package for statistical natural language processing, document classification, clustering, topic modeling, information extraction, and other machine learning applications to text.

The newest version!

/* Copyright (C) 2003 Univ. of Massachusetts Amherst, Computer Science Dept.
   This file is part of "MALLET" (MAchine Learning for LanguagE Toolkit).
   http://www.cs.umass.edu/~mccallum/mallet
   This software is provided under the terms of the Common Public License,
   version 1.0, as published by http://www.opensource.org.  For further
   information, see the file `LICENSE' included with this distribution. */





package cc.mallet.classify.tui;

import java.util.logging.*;
import java.io.*;

import cc.mallet.classify.*;
import cc.mallet.util.*;
/**
 * Diagnostic facilities for a classifier.
   @author Andrew McCallum [email protected]
 */

public class Classifier2Info
{
	private static Logger logger = MalletLogger.getLogger(Classifier2Info.class.getName());

	static CommandOption.File classifierFile = new CommandOption.File
	(Classifier2Info.class, "classifier", "FILE", true, new File("-"),
	 "Read the saved classifier from this file.", null);

	public static void main (String[] args) throws FileNotFoundException, IOException
	{
		// Process the command-line options
		CommandOption.setSummary (Classifier2Info.class,
		"A tool for printing information about saved classifiers.");
		CommandOption.process (Classifier2Info.class, args);

		// Print some helpful messages for error cases
		if (args.length == 0) {
			CommandOption.getList(Classifier2Info.class).printUsage(false);
			System.exit (-1);
		}

		// Read in the classifier
		Classifier classifier;
		try {
			ObjectInputStream ois = new ObjectInputStream (new FileInputStream (classifierFile.value));
			classifier = (Classifier) ois.readObject();
			ois.close();
		} catch (Exception e) {
			e.printStackTrace();
			throw new IllegalArgumentException ("Couldn't read classifier "+classifierFile.value);
		}

		classifier.print ();
	}


}