result_count, '' if result_count == 1 else 's')
print 'Result was {:1.2f} (0 = {}, 1 = {})'.format(
sum(results) / result_count, doctype_other, doctype_expected)
else:
print 'No documents found'
if __name__ == '__main__':
usage = 'Usage: {} <file> <expected doctype> <other doctype>'.format(
sys.argv[0])
if len(sys.argv) != 4:
raise ValueError(usage)
input_file = sys.argv[1]
doctype_expected = sys.argv[2]
doctype_other = sys.argv[3]
classifier.set_doctypes(doctype_expected, doctype_other)
results = None
if os.path.isfile(input_file):
results = check_file(input_file)
elif os.path.isdir(input_file):
results = check_dir(input_file)
else:
raise ValueError("Unable to find file/directory '{}'\n{}".format(
input_file, usage))
show_results(results)
def show_results(results):
result_count = len(results)
if result_count:
print 'Tested with %s document%s' % (result_count, '' if result_count == 1 else 's')
print 'Result was %1.2f (0 = %s, 1 = %s)' % (sum(results) / result_count, doctype_other, doctype_expected)
else :
print 'No documents found'
if __name__ == '__main__':
usage = 'Usage: %s <file> <expected doctype> <other doctype>' % sys.argv[0]
if len(sys.argv) != 4:
raise ValueError(usage)
input_file = sys.argv[1]
doctype_expected = sys.argv[2]
doctype_other = sys.argv[3]
classifier.set_doctypes(doctype_expected, doctype_other)
results = None
if os.path.isfile(input_file):
results = check_file(input_file)
elif os.path.isdir(input_file):
results = check_dir(input_file)
else:
raise ValueError("Unable to find file/directory '%s'\n%s" % (input_file, usage))
show_results(results)
