public static void main(String[] args) {
        if (args.length < 3)
            System.out.println("Wrong arguments: (-train|-annotate) corpusFileName modelFileName [annotatedCorpusFileName])");
        else {

            ClassifierNB classifier = new ClassifierNB();
            try {
                String corpusPath = args[1];
                String modelFileName = args[2];
                File corpusFile = new File(corpusPath);
                classifier.corpus = readCorpusData(corpusFile);

                if (args[0].equals("-train")) {
                    classifier.trainModel(modelFileName);
                } else if (args[0].equals("-annotate")) {
                    if (args.length < 4)
                        System.out.println("Wrong arguments: -annotate corpusFileName modelFileName annotatedCorpusFileName)");
                    else {
                        String annotatedCorpusFileName = args[3];
                        classifier.corpus.deleteAnnotation();
                        classifier.startClassifying(modelFileName, annotatedCorpusFileName);
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }