diff --git a/code/binary_bpr/main.py b/code/binary_bpr/main.py index 88c9d7d2cc4cc6071350c0c5f5e58f4bdb6b99a1..c9179a8adb1ab1e8c105fcb85dde2d41e63a3a36 100644 --- a/code/binary_bpr/main.py +++ b/code/binary_bpr/main.py @@ -62,7 +62,7 @@ def read_file(dataTrain, dataTest): dico_items = { k:v for (k,v) in zip(items, range(num_items))} return dico_kc, dico_users, dico_items - def save_embeddings(xpName: str, modelName: str, embeddings,userEmbDir : str,itemEmbDir : str): +def save_embeddings(xpName: str, modelName: str, embeddings,userEmbDir : str,itemEmbDir : str): """ Saves all the metrics measured after the training process.