Upload 2 files

nextus_regressor.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b37302fa29dadfe2628c43ccabedc0c5c81dc068852a401cb57166cce1e92ff0
+size 457056444

nextus_regressor_class.py

@@ -0,0 +1,23 @@
+import torch
+from torch import nn
+from sentence_transformers import SentenceTransformer
+from regressor import *
+import numpy as np
+
+
+class NextUsRegressor(nn.Module):
+    def __init__(self):
+        super(NextUsRegressor, self).__init__()
+        self.embedder = SentenceTransformer(ENCODER)
+        
+        
+        self.regressor = WRegressor()
+
+        return
+    
+    def forward(self, txts):
+        # expects a list of strings
+        embedded = self.embedder.encode(np.array(txts))
+        embedded_tensor = torch.tensor(embedded, dtype=torch.float32)
+        return self.regressor(embedded_tensor)
+