dash-asg / ASG.API /ASGModels.py
wasmdashai's picture
Update ASG.API/ASGModels.py
0dead86 verified
from ModelTEC import *
import pickle
from TypeModels import TypeModel
from BasePath import BASEPATH
from threading import Thread
from MScenoariotree import *
from AModelScenoariotree import ModelScenoariotree
from MCheakscenario import Cheakscenario
class ASG:
def readobsoft(self):
self.Soft=TEC(typemodel="svmL",model_setting=TypeModel['software'])
self.Soft.loadmodel()
self.Index+=1
def readobgroup(self):
self.Group=TEC(typemodel="svmL",model_setting=TypeModel['group'])
self.Group.loadmodel()
self.Index+=1
def readobtec(self):
self.Tec=TEC(typemodel="svmL",model_setting=TypeModel['technique'])
self.Tec.loadmodel()
self.Index+=1
def readmstree(self):
self.SGT=Scenoariotree('TTs.pkl','tecbyTectics.pkl')
self.SGT.Fit()
self.Index+=1
def __init__(self,isForm=True):
self.Index=0
self.SGT=None
# Thread(target=self.readmstree).start()
# Thread(target=self.readobsoft).start()
# Thread(target=self.readobtec).start()
# Thread(target=self.readobgroup).start()
self.readmstree()
self.readobsoft()
self.readobtec()
self.readobgroup()
self.Soft.setPipeline(model=self.Tec)
self.Group.setPipeline(model=self.Soft)
self.cks=Cheakscenario(ob=self.Soft)
self.MST=ModelScenoariotree(obTECSoft=self.Soft,obTEC=self.Tec,obG=self.Group,Base=self.SGT,isForm=isForm)
self.Tec.obMP
self.SGT.setModels(ObTEC=self.Tec,ObTECSodft=self.Soft,obchk=self.cks,isForm=False)
def search(self,inputstate=[''],rateerror=0,PrintFuri=True,type_search='Max',ThresholdTechnique=0.5,istrans=False,numstop=-1):
Cheakscenario.PrintFuri=PrintFuri
Cheakscenario.rateerror=rateerror
outputs=self.SGT.Predict(Description=inputstate,WF=type_search,ThresholdTechnique=ThresholdTechnique,istrans=istrans,numstop=numstop)
print ('all scenario : end')
return outputs