model added

model.py

@@ -0,0 +1,26 @@
+from datasets import load_dataset
+from sentence_transformers.losses import CosineSimilarityLoss
+
+from setfit import SetFitModel, SetFitTrainer
+
+dataset = load_dataset("ayakiri/wolo-app-categories-to-description")
+
+train_ds = dataset["train"].shuffle(seed=42).select(range(8 * 2))
+test_ds = dataset["test"]
+
+model = SetFitModel.from_pretrained("sentence-transformers/paraphrase-mpnet-base-v2")
+
+trainer = SetFitTrainer(
+    model=model,
+    train_dataset=train_ds,
+    eval_dataset=test_ds,
+    loss_class=CosineSimilarityLoss,
+    batch_size=16,
+    num_iterations=20,
+    num_epochs=1
+)
+
+trainer.train()
+metrics = trainer.evaluate()
+
+trainer.push_to_hub("ayakiri/wolo-app-categories-setfit-model")

runs/Feb01_12-42-17_DESKTOP-S8RJVAJ/events.out.tfevents.1706787738.DESKTOP-S8RJVAJ

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

runs/Feb01_12-47-52_DESKTOP-S8RJVAJ/events.out.tfevents.1706788072.DESKTOP-S8RJVAJ

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