from load_dataset import load_data
from trainer import BinaryTuner
import pandas as pd
import numpy as np
import warnings
warnings.filterwarnings("ignore")

_, dms2, dms3, _ = load_data()

mody2 = BinaryTuner(dms2, 'MODY2_label', seeds=[231964], drop_ratio=0.2)
mody2.fit()
mody2.explain_model('GaussianNB', 'fulldataset-oversampled-mice', 231964)
mody2.wrap_and_save()


mody3 = BinaryTuner(dms3, 'MODY3_label', seeds=[536202], drop_ratio=0.2)
mody3.fit()
mody3.explain_model('RandomForestClassifier', 'fulldataset-original-mice', 536202)
mody3.wrap_and_save()