Example #1
0
    def do_roc_train_test(self):
        if self.step_done("roc_train_test"):
            return

        self.do_train()

        self.logger.info("Make ROC for train and test")
        roc_train_test(self.p_classname, self.p_class, self.df_xtrain,
                       self.df_ytrain, self.df_xtest, self.df_ytest,
                       self.s_suffix, self.dirmlplot, self.p_binmin,
                       self.p_binmax)
Example #2
0
 def do_roc_train_test(self):
     roc_train_test(self.p_classname, self.p_class, self.df_xtrain,
                    self.df_ytrain, self.df_xtest, self.df_ytest,
                    self.s_suffix, self.dirmlplot)