PoliBa-DeepLearning/NanoSocrates
2
0
Fork 0
Code Issues 3 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fixed a masking problem

Browse Source
This commit is contained in:
Christian Risi 2025-10-14 10:34:14 +02:00
parent 80fd7fd600
commit 4968d79403
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Project_Model/Libs/Batch/Classes/Batcher.py
View File

@ -157,7 +157,7 @@ class Batcher:
X = [] X = []
Y = [] Y = []
for rdf in batch["RDFs"]: for rdf in batch["RDFs"]:
x, y = self._masker.mask_sequence(rdf) x, y = self._masker.mask_sequence(rdf[:self.__max_length])
X.append(x) X.append(x)
Y.append(y) Y.append(y)
return self.__normalization(X, Y) return self.__normalization(X, Y)