Update ASG.API/ModelTEC.py

ASG.API/ModelTEC.py

@@ -19,9 +19,20 @@ from sklearn.metrics import confusion_matrix, classification_report, accuracy_sc
 
 from googletrans import Translator
 
-from django.urls import ModelPums
+
 import requests
 
+class  ModelPums:
+    def __init__(self):
+        self.obData=None
+        self.obVec=None
+        self.obSVML=None
+        self.obSVMK=None
+        self.oblogict=None
+        self.X=None
+        self.Y=None
+        self.Classes=None
+    
 class TEC:
     __Outs={'svmL':None,'svmK':None,'logstick':None,'target':None}
 
@@ -250,6 +261,20 @@ class TEC:
         except:
             txttec='No'
         return txttec
+    
+    def predict_ids(self,description,is_input=False,istrans=False):
+        if istrans:
+            description=self.to_tran(description)
+
+       # clean_description,_=self.clean_data([(description,' ')],'',is_input=is_input)
+
+        try:
+            features=self.obMP.obVec.transform(description)
+            yp=self.Model.predict(features)
+          #  txttec=self.obMP.Classes[yp[0]]
+        except:
+            yp='No'
+        return yp
 
     def ptedict_proba(self,description,mx=[],is_input=False,istrans=False):
         if istrans: