Upload 6 files

config.json

@@ -0,0 +1,12 @@
+{
+  "model_type": "simple_classifier",
+  "input_dim": 200,
+  "num_classes": 3,
+  "p_dropout": 0.3,
+  "classes": [
+    "0",
+    "1",
+    "2"
+  ],
+  "transformers_version": "4.53.3"
+}

label_encoder.pkl

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

model.py

@@ -0,0 +1,20 @@
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+
+class SimpleClassifier(nn.Module):
+    def __init__(self, input_dim, num_classes, p_dropout=0.3):
+        super().__init__()
+        self.linear1 = nn.Linear(input_dim, 256)
+        self.ln1 = nn.LayerNorm(256)
+        self.dropout = nn.Dropout(p_dropout)
+        self.linear2 = nn.Linear(256, 128)
+        self.ln2 = nn.LayerNorm(128)
+        self.linear_out = nn.Linear(128, num_classes)
+
+    def forward(self, x):
+        x = F.gelu(self.ln1(self.linear1(x)))
+        x = self.dropout(x)
+        x = F.gelu(self.ln2(self.linear2(x)))
+        x = self.dropout(x)
+        return self.linear_out(x)

pytorch_model.bin

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

svd.pkl

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

vectorizer.pkl

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