Пример #1
0
    def generate_report(self, name=None, level=None, notes=None):
        self.reporting = ClassificationReporting(self.clf.best_estimator_,
                                                 self.X_train, self.X_test,
                                                 self.y_train, self.y_test,
                                                 self.n_classes)

        if name:
            self.reporting.set_name(name)
        if level:
            self.reporting.set_level(level)
        if notes:
            self.reporting.set_notes(notes)

        self.report = self.reporting.create_report(output=True, show_roc=False)

        # incase it's a function...
        self.clf.best_params_["features__text__tfidf__tokenizer"] = \
        pickle.dumps(self.clf.best_params_["features__text__tfidf__tokenizer"])

        self.report["params"] = self.clf.best_params_
        return self