import logging

import matplotlib.pyplot as plt
from sklearn.datasets import make_classification

from mla.tsne import TSNE

logging.basicConfig(level=logging.DEBUG)

X, y = make_classification(
    n_samples=500, n_features=10, n_informative=5, n_redundant=0, random_state=1111, n_classes=2, class_sep=2.5
)

p = TSNE(2, max_iter=500)
X = p.fit_transform(X)

colors = ["red", "green"]
for t in range(2):
    t_mask = (y == t).astype(bool)
    plt.scatter(X[t_mask, 0], X[t_mask, 1], color=colors[t])

plt.show()
