- Python == 3.6.13
- pytorch == 1.7.1
- numpy == 1.19.5
- scikit-learn == 0.24.2
- epochs: the number of epochs to train
- lr: learning rate
- embed_dim: embedding dimension
- N: a parameter of L0, the default is the number of triples
- droprate: dropout rate
- batch_size: batch size for training