def main():
parser = argparse.ArgumentParser()
parser.add_argument('dir', help='Path to output directory')
parser.add_argument('--t', '-test', action='store_true', help='test')
args = parser.parse_args()
print('loading data')
df = pd.read_csv('data/test.csv', delimiter=',') if args.t else pd.read_csv('data/train.csv', delimiter=',')
value = 3
if value == 1:
model = Model()
dp = model.DP()
df_processed = dp.process(df, args.t)
elif value == 2:
model = LGBMModel()
dp = model.DP()
df_processed = dp.process(df, args.t)
else:
model = GruAttModel()
dp = model.DP()
df_processed = dp.process(df, args.t)
if not args.t:
model.train(df_processed)
else:
df = model.test(df_processed)
sample_submission = pd.read_csv('data/sample_submission.csv')
if not os.path.isdir(args.dir):
os.mkdir(args.dir)
sample_submission['project_is_approved'] = df['project_is_approved']
sample_submission.to_csv(os.path.join(args.dir, 'result.csv'), index=False)
