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