Files
bayesclass/examples/plot_transformer.py
2022-11-04 11:53:06 +01:00

27 lines
658 B
Python

"""
=============================
Plotting Template Transformer
=============================
An example plot of :class:`bayesclass.template.TemplateTransformer`
"""
import numpy as np
from matplotlib import pyplot as plt
from bayesclass import TemplateTransformer
X = np.arange(50, dtype=np.float).reshape(-1, 1)
X /= 50
estimator = TemplateTransformer()
X_transformed = estimator.fit_transform(X)
plt.plot(X.flatten(), label="Original Data")
plt.plot(X_transformed.flatten(), label="Transformed Data")
plt.title("Plots of original and transformed data")
plt.legend(loc="best")
plt.grid(True)
plt.xlabel("Index")
plt.ylabel("Value of Data")
plt.show()