from constants import *
from types import SimpleNamespace
from classifier import classifier_run
from .finetuned_bert import get_finetuned_bert

ARGUMENTS = SimpleNamespace(
    dataset='sst',
    batch_size=BATCH_SIZE_SST,
    train=IDS_SST_TRAIN,
    dev=IDS_SST_DEV,
    test=IDS_SST_TEST,
    lr=1e-5,
    fine_tune_mode='full-model'
)

bert = get_finetuned_bert('sup')
classifier_run(ARGUMENTS, bert)