Mercurial > repos > goeckslab > pycaret_predict

comparison pycaret_classification.py @ 3:ccd798db5abb draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/goeckslab/Galaxy-Pycaret commit cf47efb521b91a9cb44ae5c5ade860627f9b9030
author goeckslab
date Tue, 03 Jun 2025 19:31:06 +0000
parents 0314dad38aaa
children 1aed7d47c5ec
comparison
equal deleted inserted replaced
2:0314dad38aaa 3:ccd798db5abb
1 import logging 1 import logging
2 2
3 from base_model_trainer import BaseModelTrainer 3 from base_model_trainer import BaseModelTrainer
4
5 from dashboard import generate_classifier_explainer_dashboard 4 from dashboard import generate_classifier_explainer_dashboard
6
7 from pycaret.classification import ClassificationExperiment 5 from pycaret.classification import ClassificationExperiment
8
9 from utils import add_hr_to_html, add_plot_to_html, predict_proba 6 from utils import add_hr_to_html, add_plot_to_html, predict_proba
10 7
11 LOG = logging.getLogger(__name__) 8 LOG = logging.getLogger(__name__)
12 9
13 10
62 plot_kwargs={ 59 plot_kwargs={
63 'micro': False, 60 'micro': False,
64 'macro': False, 61 'macro': False,
65 'per_class': False, 62 'per_class': False,
66 'binary': True 63 'binary': True
67 } 64 })
68 )
69 self.plots[plot_name] = plot_path 65 self.plots[plot_name] = plot_path
70 continue 66 continue
71 67
72 plot_path = self.exp.plot_model(self.best_model, 68 plot_path = self.exp.plot_model(self.best_model,
73 plot=plot_name, save=True) 69 plot=plot_name, save=True)