numfolds = 0
for i in train_test_files:
outname = '%s.%s' % (outprefix, i)
cont = args.cont
results = train_and_test_model(args, train_test_files[i], outname, cont)
if args.prefix2:
test, train, test2, train2 = results
else:
test, train = results
#write out the final predictions for test and train sets
if test.aucs:
write_results_file('%s.auc.finaltest' % outname, test.y_true, test.y_score, footer='AUC %f\n' % test.aucs[-1])
write_results_file('%s.auc.finaltrain' % outname, train.y_true, train.y_score, footer='AUC %f\n' % train.aucs[-1])
if test.rmsds:
write_results_file('%s.rmsd.finaltest' % outname, test.y_aff, test.y_predaff, footer='RMSD %f\n' % test.rmsds[-1])
write_results_file('%s.rmsd.finaltrain' % outname, train.y_aff, train.y_predaff, footer='RMSD %f\n' % train.rmsds[-1])
if test.rmsd_rmses:
write_results_file('%s.rmsd_rmse.finaltest' % outname, test.rmsd_true, test.rmsd_pred, footer='RMSE %f\n' % test.rmsd_rmses[-1])
write_results_file('%s.rmsd_rmse.finaltrain' % outname, train.rmsd_true, train.rmsd_pred, footer='RMSE %f\n' % train.rmsd_rmses[-1])
if args.prefix2:
if test2.aucs:
write_results_file('%s.auc.finaltest2' % outname, test2.y_true, test2.y_score, footer='AUC %f\n' % test2.aucs[-1])
write_results_file('%s.auc.finaltrain2' % outname, train2.y_true2, train2.y_score, footer='AUC %f\n' % train2.aucs[-1])
else:
print "Error parsing", checkname
sys.exit(1)
results = train_and_test_model(args, train_test_files[i], outname,
cont)
if args.prefix2:
test, train, test2, train2 = results
else:
test, train = results
#write out the final predictions for test and train sets
if test.aucs:
write_results_file('%s.auc.finaltest' % outname,
test.y_true,
test.y_score,
footer='AUC %f\n' % test.aucs[-1])
write_results_file('%s.auc.finaltrain' % outname,
train.y_true,
train.y_score,
footer='AUC %f\n' % train.aucs[-1])
if test.rmsds:
write_results_file('%s.rmsd.finaltest' % outname,
test.y_aff,
test.y_predaff,
footer='RMSD %f\n' % test.rmsds[-1])
write_results_file('%s.rmsd.finaltrain' % outname,
train.y_aff,
train.y_predaff,
footer='RMSD %f\n' % train.rmsds[-1])