From 0b3408d475c0ee6fbe810fa6e2a51b39de42105b Mon Sep 17 00:00:00 2001 From: Meraldo Antonio Date: Sat, 5 Oct 2024 23:32:53 +0800 Subject: [PATCH] Uncommented visualize model --- skpro/regression/bayesian.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/skpro/regression/bayesian.py b/skpro/regression/bayesian.py index fb5a1701..7abfd2a2 100644 --- a/skpro/regression/bayesian.py +++ b/skpro/regression/bayesian.py @@ -9,8 +9,7 @@ __author__ = ["meraldoantonio"] from skpro.regression.base import BaseProbaRegressor - -# from skpro.utils.validation._dependencies import _check_soft_dependencies +from skpro.utils.validation._dependencies import _check_soft_dependencies class BayesianLinearRegressor(BaseProbaRegressor): @@ -179,16 +178,16 @@ def _fit(self, X, y): self.idata.add_groups(training_data=training_data.to_xarray()) return self - # def visualize_model(self, **kwargs): - # """Use Graphviz to visualize the model flow.""" - # _check_soft_dependencies( - # "graphviz", msg="You need to install Graphviz to use this method!" - # ) - # import pymc as pm + def visualize_model(self, **kwargs): + """Use Graphviz to visualize the model flow.""" + _check_soft_dependencies( + "graphviz", msg="You need to install Graphviz to use this method!" + ) + import pymc as pm - # assert self._is_fitted, "You need to fit the model before visualizing it!" + assert self._is_fitted, "You need to fit the model before visualizing it!" - # return pm.model_to_graphviz(self.model, **kwargs) + return pm.model_to_graphviz(self.model, **kwargs) def _sample_dataset(self, group_name, return_type=None): """