baseline.py

print(" Train Data : ")
print("#"*100)
randomlist = np.random.rand(train_data.shape[0])
pred_y = [ 1 if val > 0.5 else 0 for val in randomlist]
print(" Train AUC Score Random Model : ", roc_auc_score(train_labels[:, 0], pred_y))
print(" Train F1 - Score Random Model : ", f1_score(train_labels[:, 0], pred_y))
plot_confusion_matrix(train_labels[:, 1], pred_y)

print(" CV Data : ")
print("#"*100)
randomlist = np.random.rand(cv_data.shape[0])
pred_y = [ 1 if val > 0.5 else 0 for val in randomlist]
print(" Test AUC Score Random Model : ", roc_auc_score(cv_labels[:, 0], pred_y))
print(" Test F1 - Score Random Model : ", f1_score(cv_labels[:, 0], pred_y))
plot_confusion_matrix(cv_labels[:, 1], pred_y)