# Test driver
import numpy as np
from Tasks import *
q = Tasks(full_dataset_filename='./dataset/Training_Data.csv',
judge_filename='dataset/Training_Data_9col.csv')
one_hot_dataset = q.Q_02(q.judge_dataset)
q.EDA(one_hot_dataset)
# np.set_printoptions(threshold=np.inf)
# y_pred = q.Q_17(q.judge_dataset)
# print("y pred: \n", y_pred)
# --- KEV MAIN ---
# print("\n\nQuestion 1")
# print(q.Q_01())
#
#
# print("\n\nQuestion 2")
# one_hot_dataset = q.Q_02(q.full_dataset)
# print(one_hot_dataset)
#
#
# print("\n\nQuestion 3")
# missing_count, revised_full_dataset = q.Q_03(one_hot_dataset)
# print(missing_count)
