... TorchFM, model_params={'n': X_train.shape[1], 'k': 5}, batch_size=1024, epochs=300, criterion=nn.BCEWithLogitsLoss(), optimizer_class=torch.optim.SGD ...
確定! 回上一頁