wasmdashai commited on
Commit
2ba98ce
1 Parent(s): b8284a1

Update ASG.API/ModelTEC.py

Browse files
Files changed (1) hide show
  1. ASG.API/ModelTEC.py +2 -0
ASG.API/ModelTEC.py CHANGED
@@ -61,6 +61,8 @@ class TEC:
61
  matrck=pickle.load(open(self.model_setting.path_model,'rb'))
62
  self.DES=pickle.load(open(self.model_setting.path_Qwords,'rb'))
63
  mm.obData=matrck['obData']
 
 
64
  mm.obVec=matrck['obVec']
65
  mm.obSVML=matrck['obSVML']
66
  mm.obSVMK=matrck['obSVMK']
 
61
  matrck=pickle.load(open(self.model_setting.path_model,'rb'))
62
  self.DES=pickle.load(open(self.model_setting.path_Qwords,'rb'))
63
  mm.obData=matrck['obData']
64
+ mm.obVec=TfidfVectorizer(norm='l2')
65
+ mm.obVec.fit(mm.obData)
66
  mm.obVec=matrck['obVec']
67
  mm.obSVML=matrck['obSVML']
68
  mm.obSVMK=matrck['obSVMK']