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

oikan/metrics.py

@@ -0,0 +1,25 @@
+import numpy as np
+import torch
+
+def evaluate_regression(model, X, y):
+    # Evaluate regression performance (MSE, MAE, RMSE)
+    with torch.no_grad():
+        X_tensor = torch.FloatTensor(X)
+        y_pred = model(X_tensor).numpy().ravel()
+    mse = np.mean((y - y_pred)**2)
+    mae = np.mean(np.abs(y - y_pred))
+    rmse = np.sqrt(mse)
+    print("Mean Squared Error:", mse)
+    print("Mean Absolute Error:", mae)
+    print("Root Mean Squared Error:", rmse)
+    return mse, mae, rmse
+
+def evaluate_classification(model, X, y):
+    # Evaluate classification accuracy
+    with torch.no_grad():
+        X_tensor = torch.FloatTensor(X)
+        logits = model(X_tensor)
+        y_pred = torch.argmax(logits, dim=1).numpy()
+    accuracy = np.mean(y_pred == y)
+    print("Classification Accuracy:", accuracy)
+    return accuracy

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.7.dist-info}/METADATA

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

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

@@ -1,12 +1,13 @@
 oikan/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+oikan/metrics.py,sha256=z3V0GI_edR7hYt_ic08Utfln5ynfqCfO1Gq0Yq4Q42Q,846
 oikan/model.py,sha256=9_U3jh1YwASbLOgHpFm4F80J3QGEhzIgQHNkqbZCPJs,2920
 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.7.dist-info/LICENSE,sha256=75ASVmU-XIpN-M4LbVmJ_ibgbzbvRLVti8FhnR0BTf8,1096
+oikan-0.0.1.7.dist-info/METADATA,sha256=onbMLORaKPQnqA-nZmydvoLDSR0xtJfh44BqWf_n2Ls,2962
+oikan-0.0.1.7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+oikan-0.0.1.7.dist-info/top_level.txt,sha256=XwnwKwTJddZwIvtrUsAz-l-58BJRj6HjAGWrfYi_3QY,6
+oikan-0.0.1.7.dist-info/RECORD,,