20 lines
562 B
Python
20 lines
562 B
Python
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()
|