Update model.py

model.py

@@ -42,11 +42,14 @@ class FleshkaTabularTransformer(PreTrainedModel):
                 if module.bias is not None:
                     nn.init.zeros_(module.bias)
 
-    def forward(self, x: list):
+    def forward(self, lx: list):
         # x: [batch_size, input_dim]
-        x = tensor(self._normalize(x), dtype=float32).unsqueeze(0)
-        x = self.input_proj(x)           # [batch_size, d_model]
-        x = x.unsqueeze(1)               # [batch_size, 1, d_model] (добавляем seq_len=1)
-        x = self.transformer(x)          # [batch_size, 1, d_model]
-        x = x.squeeze(1)                 # [batch_size, d_model]
-        return self.head(x)              # [batch_size, 1]
+        out = list()
+        for nx in lx:
+            x = tensor(self._normalize(nx), dtype=float32).unsqueeze(0)
+            x = self.input_proj(x)           # [batch_size, d_model]
+            x = x.unsqueeze(1)               # [batch_size, 1, d_model] (добавляем seq_len=1)
+            x = self.transformer(x)          # [batch_size, 1, d_model]
+            x = x.squeeze(1)                 # [batch_size, d_model]
+            out.append(self.head(x).item().item() > 0)
+        return out