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")