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('unsup') | |
classifier_run(ARGUMENTS, bert) |