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

ARGUMENTS = SimpleNamespace(
    dataset='cfimdb',
    batch_size=BATCH_SIZE_CFIMDB,
    train=IDS_CFIMDB_TRAIN,
    dev=IDS_CFIMDB_DEV,
    test=IDS_CFIMDB_TEST,
    lr=1e-5,
    fine_tune_mode='full-model'
)

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