| 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) |