"""
=============================
Plotting Template Transformer
=============================

An example plot of :class:`tclf.template.TemplateTransformer`
"""
import numpy as np
from matplotlib import pyplot as plt
from tclf import TemplateTransformer

X = np.arange(50, dtype=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()