--- a +++ b/view_results.py @@ -0,0 +1,15 @@ +import argparse +import pandas as pd + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("id", type=int) + parser.add_argument("--features", action="store_true") + args = parser.parse_args() + + results = pd.read_csv('/home/ubuntu/experiments/results.csv') + row = results[results['Id'] == args.id].iloc[0] + print row + if args.features: + print row['Important Features'].replace("\\n", "\n")