oikan 0.0.1.5__py3-none-any.whl → 0.0.1.6__py3-none-any.whl

oikan/visualize.py

@@ -17,21 +17,53 @@ def visualize_regression(model, X, y):
 def visualize_classification(model, X, y):
     model.eval()
     
-    # Create a mesh grid
-    x_min, x_max = X[:, 0].min() - 1, X[:, 0].max() + 1
-    y_min, y_max = X[:, 1].min() - 1, X[:, 1].max() + 1
-    xx, yy = np.meshgrid(np.linspace(x_min, x_max, 100),
-                        np.linspace(y_min, y_max, 100))
-    
-    # Make predictions
-    with torch.no_grad():
-        X_grid = torch.FloatTensor(np.c_[xx.ravel(), yy.ravel()])
-        Z = model(X_grid)
-        Z = torch.argmax(Z, dim=1).numpy()
-        Z = Z.reshape(xx.shape)
-    
-    # Plot
-    plt.figure(figsize=(10, 8))
-    plt.contourf(xx, yy, Z, alpha=0.4)
-    plt.scatter(X[:, 0], X[:, 1], c=y, alpha=0.8)
-    plt.show()
+    if X.shape[1] > 2:
+        # SVD projection for high-dimensional inputs.
+        X_mean = np.mean(X, axis=0)
+        X_centered = X - X_mean
+        _, _, Vt = np.linalg.svd(X_centered, full_matrices=False)
+        principal = Vt[:2]  # shape: (2, D)
+        X_proj = (X - X_mean) @ principal.T
+        
+        x1, x2 = X_proj[:, 0], X_proj[:, 1]
+        x_min, x_max = x1.min() - 1, x1.max() + 1
+        y_min, y_max = x2.min() - 1, x2.max() + 1
+        xx, yy = np.meshgrid(np.linspace(x_min, x_max, 100),
+                             np.linspace(y_min, y_max, 100))
+        grid_2d = np.c_[xx.ravel(), yy.ravel()]
+        # Inverse transform grid points to original space.
+        X_grid = X_mean + grid_2d @ principal
+        
+        with torch.no_grad():
+            X_grid_tensor = torch.FloatTensor(X_grid)
+            Z = model(X_grid_tensor)
+            Z = torch.argmax(Z, dim=1).numpy()
+            Z = Z.reshape(xx.shape)
+        
+        plt.figure(figsize=(10, 8))
+        plt.contourf(xx, yy, Z, alpha=0.4)
+        plt.scatter(X_proj[:, 0], X_proj[:, 1], c=y, alpha=0.8)
+        plt.title("Classification Visualization (SVD Projection)")
+        plt.show()
+        
+    else:
+        x1 = X[:, 0]
+        x2 = X[:, 1]
+        x_min, x_max = x1.min() - 1, x1.max() + 1
+        y_min, y_max = x2.min() - 1, x2.max() + 1
+        xx, yy = np.meshgrid(np.linspace(x_min, x_max, 100),
+                             np.linspace(y_min, y_max, 100))
+        grid_2d = np.c_[xx.ravel(), yy.ravel()]
+        X_grid = grid_2d
+
+        with torch.no_grad():
+            X_grid_tensor = torch.FloatTensor(X_grid)
+            Z = model(X_grid_tensor)
+            Z = torch.argmax(Z, dim=1).numpy()
+            Z = Z.reshape(xx.shape)
+        
+        plt.figure(figsize=(10, 8))
+        plt.contourf(xx, yy, Z, alpha=0.4)
+        plt.scatter(X[:, 0], X[:, 1], c=y, alpha=0.8)
+        plt.title("Classification Visualization")
+        plt.show()

{oikan-0.0.1.5.dist-info → oikan-0.0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: oikan
-Version: 0.0.1.5
+Version: 0.0.1.6
 Summary: OIKAN: Optimized Interpretable Kolmogorov-Arnold Networks
 Author: Arman Zhalgasbayev
 License: MIT

{oikan-0.0.1.5.dist-info → oikan-0.0.1.6.dist-info}/RECORD

@@ -4,9 +4,9 @@ oikan/regularization.py,sha256=D0Xc2lr5X5ORdA5ltvWDbNDuN8z0hkyoGzFo7pum2XE,1033
 oikan/symbolic.py,sha256=K1aI5JEPgKFu8dyjXxWDA-UZm8Gvfp0lU1M7c2NAPLY,5517
 oikan/trainer.py,sha256=itFCHSR_T6KHqa0D5RLRCmqFHa4lUIamsFGWKHmUZuI,1258
 oikan/utils.py,sha256=XwY6pgAgfYlUI9SOjdop91wh0_t6LfPLCiHretlw2Wg,1754
-oikan/visualize.py,sha256=8Dlk-tsqGZb63NyZBpZsLSlcsC51m2nXblQaS2Jf1y0,1142
-oikan-0.0.1.5.dist-info/LICENSE,sha256=75ASVmU-XIpN-M4LbVmJ_ibgbzbvRLVti8FhnR0BTf8,1096
-oikan-0.0.1.5.dist-info/METADATA,sha256=yDZjsC6Mh4vL-gY5EVlOYWtQ3VWSAChfu2SGJ5dS0dQ,2962
-oikan-0.0.1.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-oikan-0.0.1.5.dist-info/top_level.txt,sha256=XwnwKwTJddZwIvtrUsAz-l-58BJRj6HjAGWrfYi_3QY,6
-oikan-0.0.1.5.dist-info/RECORD,,
+oikan/visualize.py,sha256=7GTvfeFxrezHKMM9QmYaax75rMnTxMH0KYi_fbAw4-M,2501
+oikan-0.0.1.6.dist-info/LICENSE,sha256=75ASVmU-XIpN-M4LbVmJ_ibgbzbvRLVti8FhnR0BTf8,1096
+oikan-0.0.1.6.dist-info/METADATA,sha256=GOWOqXAAezQYOHMgxFn8sI2wlJotgesFA_8Yaiu2XS8,2962
+oikan-0.0.1.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+oikan-0.0.1.6.dist-info/top_level.txt,sha256=XwnwKwTJddZwIvtrUsAz-l-58BJRj6HjAGWrfYi_3QY,6
+oikan-0.0.1.6.dist-info/RECORD,,