PyTorchCML is a library of PyTorch implementations of matrix factorization (MF) and collaborative metric learning (CML), algorithms used in recommendation systems and data mining.
The embedding vectors are passed as a dictionary type, and each embedding is given a name.
I also introduced a dictionary called column_names as a member variable of the trainer for batches, so that I can set what each row of the batch represents.
The embedding vectors are passed as a dictionary type, and each embedding is given a name. I also introduced a dictionary called column_names as a member variable of the trainer for batches, so that I can set what each row of the batch represents.