def main():
# Print versions
print('The nltk version is {}.'.format(nltk.__version__))
print('The scikit-learn version is {}.'.format(sklearn.__version__))
# Initialise label class
Label.initialise_class()
# Functionality based on settings in Main class
if Main.LOAD_FORMATTED_FILES:
Main.load_formatted_files()
elif Main.LOAD_UNFORMATTED_FILES:
Main.load_unformatted_files()
# Print info
if Main.PRINT_EXCEPTION_DATA:
LegalDoc.print_exception_data()
if Main.PRINT_JUDGE_DATA:
Judge.print_all()
# Labels
if Main.AUTO_REGEX_LABEL_FILES:
Label.auto_regex_label_all_files()
if Main.ADD_MANUAL_LABELS:
ManualLabels.add_manual_labels()
if Main.SPLIT_AND_WRITE[0]:
Label.split_and_write(Main.SPLIT_AND_WRITE[1], Main.SPLIT_AND_WRITE[2],
Main.SPLIT_AND_WRITE[3], Main.SPLIT_AND_WRITE[4],
Main.SPLIT_AND_WRITE[5])
if Main.WRITE_LABELS[0]:
Label.write_labels(Main.WRITE_LABELS[1], Main.WRITE_LABELS[2])
if Main.PRINT_ALL_LABELS:
Label.print_all_tree()
if Main.PRINT_DOCS_BY_REGEX[0]:
LegalDoc.get_docs_by_regex(Main.REGEX, Main.PRINT_DOCS_BY_REGEX[1])
# Test code
if Main.RUN_TEST:
Main.test()
if Main.RUN_TEST_A_LEGAL_DOC:
Main.test_a_legal_doc()
if Main.RUN_TEST_ALL_LEGAL_DOCS:
Main.test_all_legal_docs()
# Print timers
if Main.PRINT_TIMERS:
Timer.print_all()
print("Done!")
