Upload model
Browse files- modeling_gzipembed.py +2 -2
modeling_gzipembed.py
CHANGED
@@ -35,8 +35,8 @@ class GZIPEmbeddingModel(PreTrainedModel):
|
|
35 |
return x if not return_tensor else torch.tensor(x)
|
36 |
|
37 |
def ncd_r(self,r):
|
38 |
-
i=r[0]
|
39 |
-
return i,self.ncd(r[1],r[
|
40 |
|
41 |
def normalize(self, x):
|
42 |
x = ''.join([char for char in x.lower() if char in "abcdefghijklmnopqrstuvwxyz "])
|
|
|
35 |
return x if not return_tensor else torch.tensor(x)
|
36 |
|
37 |
def ncd_r(self,r):
|
38 |
+
i=r[0][0]
|
39 |
+
return i,self.ncd(r[0][1],r[1])
|
40 |
|
41 |
def normalize(self, x):
|
42 |
x = ''.join([char for char in x.lower() if char in "abcdefghijklmnopqrstuvwxyz "])
|