jgrosjean commited on
Commit
bbab9f8
1 Parent(s): 5b5aee3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -81,14 +81,14 @@ from sklearn.metrics.pairwise import cosine_similarity
81
  sentence_1 = ["Der Zug kommt um 9 Uhr in Zürich an."]
82
  sentence_2 = ["Le train arrive à Lausanne à 9h."]
83
 
84
- #Compute embedding for both
85
  embedding_1 = generate_sentence_embedding(sentence_1, language="de")
86
  embedding_2 = generate_sentence_embedding(sentence_2, language="fr")
87
 
88
- #Compute cosine-similarity
89
  cosine_score = cosine_similarity(embedding_1, embedding_2)
90
 
91
- #Output the score
92
  print("The cosine score for", sentence_1, "and", sentence_2, "is", cosine_score)
93
  ```
94
  Output:
@@ -119,9 +119,9 @@ The fine-tuning script can be accessed [here](Link).
119
 
120
  #### Training Hyperparameters
121
 
122
- Number of epochs: 1
123
- Learning rate: 1e-5
124
- Batch size: 512
125
 
126
  ## Evaluation
127
 
 
81
  sentence_1 = ["Der Zug kommt um 9 Uhr in Zürich an."]
82
  sentence_2 = ["Le train arrive à Lausanne à 9h."]
83
 
84
+ # Compute embedding for both
85
  embedding_1 = generate_sentence_embedding(sentence_1, language="de")
86
  embedding_2 = generate_sentence_embedding(sentence_2, language="fr")
87
 
88
+ # Compute cosine-similarity
89
  cosine_score = cosine_similarity(embedding_1, embedding_2)
90
 
91
+ # Output the score
92
  print("The cosine score for", sentence_1, "and", sentence_2, "is", cosine_score)
93
  ```
94
  Output:
 
119
 
120
  #### Training Hyperparameters
121
 
122
+ - Number of epochs: 1
123
+ - Learning rate: 1e-5
124
+ - Batch size: 512
125
 
126
  ## Evaluation
127