From 8e350ad4fd5cd50dccc684352080bedd1e8f4835 Mon Sep 17 00:00:00 2001
From: Pritimay Sarkar <pritimay@sminnovations.in>
Date: Wed, 28 Feb 2024 14:43:41 +0530
Subject: [PATCH] add tsne

---
 train/tsne.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 train/tsne.py

diff --git a/train/tsne.py b/train/tsne.py
new file mode 100644
index 0000000..8c17ad5
--- /dev/null
+++ b/train/tsne.py
@@ -0,0 +1,18 @@
+import numpy as np
+import matplotlib.pyplot as plt
+from sklearn import datasets
+from sklearn.manifold import TSNE
+
+iris = datasets.load_iris()
+X = iris.data
+y = iris.target
+
+# t-SNE with two components (2D visualization)
+tsne = TSNE(n_components=2, random_state=42)
+X_tsne = tsne.fit_transform(X)
+
+plt.scatter(X_tsne[:, 0], X_tsne[:, 1], c=y, cmap=plt.cm.Set1, edgecolor='k')
+plt.title("t-SNE Visualization of Iris Dataset")
+plt.xlabel("t-SNE Component 1")
+plt.ylabel("t-SNE Component 2")
+plt.show()