Hans Elias J
commited on
Commit
•
0996185
1
Parent(s):
e66ef1e
normalize embedding vector
Browse files- handler.py +1 -0
handler.py
CHANGED
@@ -31,5 +31,6 @@ class EndpointHandler():
|
|
31 |
outputs = self.model(**batch_dict)
|
32 |
|
33 |
embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask']).tolist()
|
|
|
34 |
|
35 |
return embeddings
|
|
|
31 |
outputs = self.model(**batch_dict)
|
32 |
|
33 |
embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask']).tolist()
|
34 |
+
embeddings = F.normalize(embeddings, p=2, dim=1)
|
35 |
|
36 |
return embeddings
|