示例#1
0
文件: Bdt.py 项目: yuliasheleva/hepML
 def compareTrainTest(self, doEvalSet=False):
     if doEvalSet:
         compareTrainTest(self.bdt.decision_function,self.data.X_dev,self.data.y_dev,\
                 self.data.X_eval,self.data.y_eval,self.output,append='_eval')
     else:
         compareTrainTest(self.bdt.decision_function,self.data.X_train,self.data.y_train,\
                 self.data.X_test,self.data.y_test,self.output)
示例#2
0
文件: Dnn.py 项目: yuliasheleva/hepML
    def compareTrainTest(self, doEvalSet=False):

        if doEvalSet:
            compareTrainTest(self.model.predict,self.data.X_dev.as_matrix(),self.data.y_dev.as_matrix(),\
                self.data.X_eval.as_matrix(),self.data.y_eval.as_matrix(),self.output,append='_eval')
        else:
            compareTrainTest(self.model.predict,self.data.X_train.as_matrix(),self.data.y_train.as_matrix(),\
                self.data.X_test.as_matrix(),self.data.y_test.as_matrix(),self.output)