andrea.pedrotti
/
gFun
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

early stopping + typos

This commit is contained in:
andrea 2021-01-26 18:13:34 +01:00
parent e9a410faa4
commit 79cdaa0beb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/view_generators.py
View File

@ -370,10 +370,10 @@ class BertGen(ViewGen):
self.model.to('cuda' if self.gpus else 'cpu')
self.model.eval()
# time_init = time.time()
l_emebds = self.model.encode(data, batch_size=64)
l_embeds = self.model.encode(data, batch_size=64)
# transform_time = round(time.time() - time_init, 3)
# print(f'Executed! Transform took: {transform_time}')
return l_emebds
return l_embeds
def fit_transform(self, lX, ly):
# we can assume that we have already indexed data for transform() since we are first calling fit()