diff --git a/examples/whpa_demo.py b/examples/whpa_demo.py index e794fd8..de9ae91 100644 --- a/examples/whpa_demo.py +++ b/examples/whpa_demo.py @@ -4,7 +4,6 @@ from os.path import join as jp import pandas as pd -from loguru import logger from sklearn.cross_decomposition import CCA from sklearn.decomposition import PCA from sklearn.pipeline import Pipeline @@ -111,12 +110,6 @@ def init_bel(): # The posterior distribution is computed within the method below. y_predicted = model.predict(X_test) - # Save the fitted BEL model - # joblib.dump(model, jp(obj_dir, "bel.pkl")) - - # msg = f"model trained and saved in {obj_dir}" - # logger.info(msg) - # %% Visualization # Plot raw data @@ -129,7 +122,7 @@ def init_bel(): Y_obs=y_test, base_dir=sub_dir, ) - # + # Plot PCA pca_vision( model,