def main(args):
in_dir = join(dirname(__file__), '../../data/labeled/')
out_dir = join(dirname(__file__),
'../../output/predictions/{}'.format(args.community))
if not run_exists(out_dir, args.run_id):
logger.info('Run %s does not exists. Start new...', args.run_id)
pred = make_predictions(args)
save_predictions(pred, out_dir, args.run_id)
logger.info('Finish training and testing.')
results_test = evaluation.evaluate(
csv_true=join(in_dir, '{}_test.csv'.format(args.community)),
csv_pred=join(out_dir, '{}_test.csv'.format(args.run_id)),
label_encoder=preprocessing.LABEL_ENCODER)
evaluation.print_results(results_test, '{}_test'.format(args.run_id))
def main(args):
ground_truth = join(
dirname(__file__),
'../../data/labeled/{}_test.csv'.format(args.community))
out_dir = join(dirname(__file__),
'../../output/predictions/{}'.format(args.community))
files = glob.glob(join(out_dir, '*.csv'))
files = _filter_files(files)
results_test = evaluation.evaluate(
csv_true=ground_truth,
csv_pred=files,
label_encoder=preprocessing.LABEL_ENCODER)
report_dir = join(dirname(__file__), '../../output/evaluation/')
evaluation.save_results_csv(results_test, report_dir, args.community)
evaluation.save_results_excel(results_test, report_dir, args.community)
def main(args):
in_dir = join(dirname(__file__), '../../data/labeled/')
out_dir = join(dirname(__file__), '../../output/predictions/{}'.format(args.community))
if not run_exists(out_dir, args.run_id):
pred = make_predictions(
join(in_dir, '{}_train.csv'.format(args.community)),
join(in_dir, '{}_test.csv'.format(args.community)),
args
)
save_predictions(pred, out_dir, args.run_id)
le = LabelEncoder().fit(['CLEAR', 'UNCLEAR'])
results_test = evaluation.evaluate(
csv_true=join(in_dir, '{}_test.csv'.format(args.community)),
csv_pred=join(out_dir, '{}_test.csv'.format(args.run_id)),
label_encoder=le
)
evaluation.print_results(results_test, '{}_test'.format(args.run_id))