def explicit():
train, test = utils.get_movielens_train_test_split()
pipeline = BasePipeline(train, test=test, model=BaseModule,
n_factors=10, batch_size=1024, dropout_p=0.02,
lr=0.02, weight_decay=0.1,
optimizer=torch.optim.Adam, n_epochs=40,
verbose=True, random_seed=2017)
pipeline.fit()
def hogwild():
train, test = utils.get_movielens_train_test_split(implicit=True)
pipeline = BasePipeline(train, test=test, verbose=True,
batch_size=1024, num_workers=4,
n_factors=20, weight_decay=0,
dropout_p=0., lr=.2, sparse=True,
optimizer=torch.optim.SGD, n_epochs=40,
random_seed=2017, loss_function=bpr_loss,
model=BPRModule, hogwild=True,
interaction_class=PairwiseInteractions,
eval_metrics=('auc', 'patk'))
pipeline.fit()