def main(logger, args):
df_train, _ = load_data(INPUT_DIR, logger)
logger.info('Preprocess text')
df_train = preprocess_text(df_train)
seq_train, tokenizer = tokenize_text(df_train, logger)
logger.info('Pad train text data')
seq_train = pad_sequences(seq_train, maxlen=PADDING_LENGTH)
label_train = df_train['target'].values.reshape(-1, 1)
logger.info('Load multiple embeddings')
embedding_matrix = load_multiple_embeddings(
tokenizer.word_index,
embed_types=[0, 1, 2],
max_workers=args['max_workers'])
embedding_matrix = np.array(embedding_matrix).mean(axis=0)
# ===== training and evaluation loop ===== #
device_ids = args['device_ids']
output_device = device_ids[0]
torch.cuda.set_device(device_ids[0])
set_seed(SEED)
batch_size = args['batch_size'] * len(device_ids)
max_workers = args['max_workers']
if args['debug']:
epochs = 2
else:
epochs = EPOCHS
logger.info('Start training and evaluation loop')
skf = StratifiedKFold(n_splits=KFOLD, shuffle=True, random_state=SEED)
oof_preds = np.zeros(seq_train.shape[0])
results = []
for fold, (index_train,
index_valid) in enumerate(skf.split(label_train, label_train)):
logger.info(
f'Fold {fold + 1} / {KFOLD} - create dataloader and build model')
x_train, x_valid = seq_train[index_train].astype(
int), seq_train[index_valid].astype(int)
y_train, y_valid = label_train[index_train].astype(
np.float32), label_train[index_valid].astype(np.float32)
dataset_train = SimpleDataset(x_train, y_train)
dataset_valid = SimpleDataset(x_valid, y_valid)
dataloader_train = DataLoader(dataset=dataset_train,
batch_size=batch_size,
shuffle=True,
pin_memory=True,
worker_init_fn=worker_init_fn)
dataloader_valid = DataLoader(dataset=dataset_valid,
batch_size=batch_size,
shuffle=False,
pin_memory=True,
worker_init_fn=worker_init_fn)
dataloaders = {'train': dataloader_train, 'valid': dataloader_valid}
model = StackedNormalizedRNNFM(embedding_matrix,
PADDING_LENGTH,
hidden_size=64,
out_hidden_dim=64,
embed_drop=0.2,
out_drop=0.3,
residual=False)
model.to(output_device)
criteria = [[nn.BCEWithLogitsLoss(reduction='mean')], [1.0]]
metric = f1_from_logits_optimized
optimizer = optim.Adam(model.parameters(), lr=0.001)
scheduler = None
model_save_path = str(
DATA_DIR.joinpath(
f'models/{Path(__file__).stem}_fold_{fold}.model'))
model_name = model._get_name()
config = {
'epochs': epochs,
'loss_names': ['BCE Loss'],
'metric_type': 'batch',
'model_save_path': model_save_path,
'output_device': output_device,
'mode': 'max',
'early_stopping': 200,
'model_name': model_name,
'reg_lambda': None,
'fold': fold
}
model, valid_score, best_epoch = train_model(model, criteria, metric,
optimizer, scheduler,
dataloaders, logger,
config)
results.append({
'fold': fold,
'best_score': valid_score,
'best_epoch': best_epoch
})
message = f'Training and evaluation for the fold {fold + 1} / {KFOLD} has been done.\n'
message += f'Validation F1 score: {valid_score}\n'
logger.post(message)
dataloader_valid = DataLoader(dataset=dataset_valid,
batch_size=batch_size,
shuffle=False,
pin_memory=True)
oof_preds[index_valid] = sp.special.expit(
predict(model, dataloader_valid, config).reshape(-1, ))
logger.post(f'K-Fold train and evaluation results: {results}')
logger.info(
'Training and evaluation loop has been done. Start f1 threshold search.'
)
search_result = threshold_search(label_train.reshape(-1, ), oof_preds)
logger.post(
f'Threshold search result - f1: {search_result["f1"]}, threshold: {search_result["threshold"]}'
)
def train(embedding_matrix):
skf = StratifiedKFold(n_splits=KFOLD, shuffle=True, random_state=SEED)
oof_preds = np.zeros(seq_train.shape[0])
results = []
for fold, (index_train, index_valid) in enumerate(
skf.split(label_train, label_train)):
logger.info(
f'Fold {fold + 1} / {KFOLD} - create dataloader and build model'
)
x_train, x_valid = seq_train[index_train].astype(
int), seq_train[index_valid].astype(int)
y_train, y_valid = label_train[index_train].astype(
np.float32), label_train[index_valid].astype(np.float32)
dataset_train = SimpleDataset(x_train, y_train)
dataset_valid = SimpleDataset(x_valid, y_valid)
dataloader_train = DataLoader(dataset=dataset_train,
batch_size=batch_size,
shuffle=True,
pin_memory=True)
dataloader_valid = DataLoader(dataset=dataset_valid,
batch_size=batch_size,
shuffle=False,
pin_memory=True)
dataloaders = {
'train': dataloader_train,
'valid': dataloader_valid
}
model = StackedRNNFM(embedding_matrix,
PADDING_LENGTH,
hidden_size=64)
model.to(output_device)
criteria = [[nn.BCEWithLogitsLoss(reduction='mean')], [1.0]]
metric = f1_from_logits
optimizer = optim.Adam(model.parameters(), lr=0.001)
scheduler = None
model_save_path = str(
DATA_DIR.joinpath(
f'models/{Path(__file__).stem}_fold_{fold}.model'))
model_name = model._get_name()
config = {
'epochs': epochs,
'loss_names': ['BCE Loss'],
'metric_type': 'batch',
'model_save_path': model_save_path,
'output_device': output_device,
'mode': 'max',
'early_stopping': 200,
'model_name': model_name,
'reg_lambda': None,
'fold': fold
}
model, valid_score, best_epoch = train_model(
model, criteria, metric, optimizer, scheduler, dataloaders,
logger, config)
results.append({
'fold': fold,
'best_score': valid_score,
'best_epoch': best_epoch
})
message = f'Training and evaluation for the fold {fold + 1} / {KFOLD} has been done.\n'
message += f'Validation F1 score: {valid_score}\n'
logger.post(message)
dataloader_valid = DataLoader(dataset=dataset_valid,
batch_size=batch_size,
shuffle=False,
pin_memory=True)
oof_preds[index_valid] = sp.special.expit(
predict(model, dataloader_valid, config).reshape(-1, ))
logger.post(f'K-Fold train and evaluation results: {results}')
return oof_preds