beeformer commited on
Commit
dceae9f
·
verified ·
1 Parent(s): 7284995

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -3
README.md CHANGED
@@ -41,12 +41,10 @@ We use the pretrained [`sentence-transformers/all-mpnet-base-v2`](https://huggin
41
 
42
  We use the initial model without modifying its architecture or pre-trained model parameters.
43
  However, we reduce the processed sequence length to 384 to reduce the training time of the model.
44
- Regarding other hyperparameters, we use the same interaction data batch size of 1024; we use the negative sampling parameter m = 10000.
45
- We use constant learning rate of 1e-5, and we train the model for five epochs.
46
 
47
  ### Dataset
48
 
49
- We finetuned our model on the Goodbooks-10k dataset with item descriptions generated with [`meta-llama/Meta-Llama-3.1-8B-Instruct`](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) model. For details please see the dataset page [`beeformer/recsys-movielens-20m`](https://huggingface.co/datasets/beeformer/recsys-movielens-20m).
50
 
51
  ## Evaluation Results
52
 
 
41
 
42
  We use the initial model without modifying its architecture or pre-trained model parameters.
43
  However, we reduce the processed sequence length to 384 to reduce the training time of the model.
 
 
44
 
45
  ### Dataset
46
 
47
+ We finetuned our model on the MovieLens-20M dataset with item descriptions generated with [`meta-llama/Meta-Llama-3.1-8B-Instruct`](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) model. For details please see the dataset page [`beeformer/recsys-movielens-20m`](https://huggingface.co/datasets/beeformer/recsys-movielens-20m).
48
 
49
  ## Evaluation Results
50