public static NModelParams initModelParams(String arg, String arg2) {
NModelParams modelParams = new NModelParams();
if (arg.compareTo("S0") == 0) {
modelParams.setLineageDistance(new CLineageSimpleDistance());
} else if (arg.compareTo("SC") == 0) {
modelParams.setLineageDistance(new CLineageSimpleDistance());
modelParams.addCategories();
} else if (arg.compareTo("P0") == 0) {
modelParams.setLineageDistance(new CLineagePathDistance());
} else if (arg.compareTo("PC") == 0) {
modelParams.setLineageDistance(new CLineagePathDistance());
modelParams.addCategories();
} else if (arg.compareTo("SP") == 0) {
modelParams.setLineageDistance(new CLineageSiblingParentDistance());
}
if (arg2 != null) {
CTaxonomyModel.numClasses = Integer.valueOf(arg2);
}
return modelParams;
}
/**
* Main method for the command line application to compute the taxonomy model features using a
* Bayesian approach.
*
* @param args
*/
public static void main(String[] args) {
try {
CEnv.init();
if (args != null && args.length > 0) {
/*
* Launch the training phase.
*/
CTaxonomyClassifier classifier = null;
if (args[0].compareTo("-train") == 0) {
NModelParams modelParams = new NModelParams();
modelParams.addCategories();
CTaxonomyModel.getInstance(modelParams);
classifier = new CTaxonomyClassifier(1);
classifier.train();
}
/*
* Launch the validation phase.
*/
else if (args[0].compareTo("-validate") == 0) {
NModelParams modelParams = new NModelParams();
try {
CTaxonomyModel.init(modelParams);
classifier = new CTaxonomyClassifier(1);
classifier.validate(VALIDATION_RANGE);
int numSamples = classifier.getNumSamples();
CLogger.info(
"Taxonomy validation done with " + numSamples + " samples",
CLogger.TAXONOMY_TRAIN_TRACE);
} catch (InitException e) {
CLogger.error(e.toString());
}
}
/*
* Launch the test phase.
*/
else if (args[0].compareTo("-test") == 0) {
NModelParams modelParams = initModelParams(args[1], args[2]);
try {
CTaxonomyModel.init(modelParams);
classifier = new CTaxonomyClassifier(1);
classifier.test(TEST_RANGE);
int numSamples = classifier.getNumSamples();
CLogger.info(
"Taxonomy validation done with " + numSamples + " samples",
CLogger.TAXONOMY_TRAIN_TRACE);
} catch (InitException e) {
CLogger.error(e.toString());
}
}
} else {
CLogger.info("Command line: CTaxonomyTrainApp {-train,-validate,-all}");
}
} catch (InitException e) {
CLogger.error(e.toString());
}
}