oikan 0.0.1.9__py3-none-any.whl → 0.0.1.10__py3-none-any.whl

oikan/trainer.py

@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 from .regularization import RegularizedLoss
 
-def train(model, train_data, epochs=100, lr=0.01, save_path=None):
+def train(model, train_data, epochs=100, lr=0.01, save_path=None, verbose=True):
     '''Train regression model using MSE loss with regularization.
     Optionally save the model when training is finished if save_path is provided.
     '''
@@ -19,13 +19,13 @@ def train(model, train_data, epochs=100, lr=0.01, save_path=None):
         loss.backward()  # Backpropagate errors
         optimizer.step()  # Update parameters
         
-        if (epoch + 1) % 10 == 0:
+        if (epoch + 1) % 10 == 0 and verbose:
             print(f'Epoch [{epoch+1}/{epochs}], Loss: {loss.item():.4f}')
-    if save_path:
+    if save_path is not None:
         torch.save(model.state_dict(), save_path)
         print(f"Model saved to {save_path}")
 
-def train_classification(model, train_data, epochs=100, lr=0.01, save_path=None):
+def train_classification(model, train_data, epochs=100, lr=0.01, save_path=None, verbose=True):
     '''Train classification model using CrossEntropy loss with regularization.
     Optionally save the model when training is finished if save_path is provided.
     '''
@@ -42,8 +42,8 @@ def train_classification(model, train_data, epochs=100, lr=0.01, save_path=None)
         loss.backward()  # Backpropagation
         optimizer.step()  # Parameter update
         
-        if (epoch + 1) % 10 == 0:
+        if (epoch + 1) % 10 == 0 and verbose:
             print(f'Epoch [{epoch+1}/{epochs}], Loss: {loss.item():.4f}')
-    if save_path:
+    if save_path is not None:
         torch.save(model.state_dict(), save_path)
         print(f"Model saved to {save_path}")

{oikan-0.0.1.9.dist-info → oikan-0.0.1.10.dist-info}/METADATA

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

{oikan-0.0.1.9.dist-info → oikan-0.0.1.10.dist-info}/RECORD

@@ -3,11 +3,11 @@ oikan/metrics.py,sha256=65txPbjhTz7lCXCLtTAJTS4E5Hx42wzZ3jKar3lH_bY,860
 oikan/model.py,sha256=zlw_4HbSK3IiQhE8M4NitvrXa7vffBWsOR-HLRSJADA,3944
 oikan/regularization.py,sha256=xt8JNnPdHRAQgzF_vnyme005hWLunz9Vo2qw6m08NMM,1145
 oikan/symbolic.py,sha256=RRYHOCOCJr5KXRhdcCPvT_OqyNcCnWCWt7fOtos8rRI,5765
-oikan/trainer.py,sha256=M7F5FLCXB5HUKhMRJiX92o0DB9NHWZIb25bGgyYwXrM,1986
+oikan/trainer.py,sha256=PwA8PnVUiv5wYlQqj3DTplCAUZljZ4iWJUKUDvmIvX0,2062
 oikan/utils.py,sha256=xbVgrbhXYj57RdD3uNPchjyfmP6Kur7tngoZPa3qWOw,2094
 oikan/visualize.py,sha256=sA__nLB35y6tuWDAM3aoC7VezxJAvdSlwzmoPfrnFhQ,2249
-oikan-0.0.1.9.dist-info/LICENSE,sha256=75ASVmU-XIpN-M4LbVmJ_ibgbzbvRLVti8FhnR0BTf8,1096
-oikan-0.0.1.9.dist-info/METADATA,sha256=ZyH9i_F8ymdalTVQxVbyyEFUa0xPLt7byLXP5zVfjO0,3847
-oikan-0.0.1.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-oikan-0.0.1.9.dist-info/top_level.txt,sha256=XwnwKwTJddZwIvtrUsAz-l-58BJRj6HjAGWrfYi_3QY,6
-oikan-0.0.1.9.dist-info/RECORD,,
+oikan-0.0.1.10.dist-info/LICENSE,sha256=75ASVmU-XIpN-M4LbVmJ_ibgbzbvRLVti8FhnR0BTf8,1096
+oikan-0.0.1.10.dist-info/METADATA,sha256=Mzmf6JP3taLsxytaSz8TI30M-uvmaq7jcEkZORGCatw,3848
+oikan-0.0.1.10.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+oikan-0.0.1.10.dist-info/top_level.txt,sha256=XwnwKwTJddZwIvtrUsAz-l-58BJRj6HjAGWrfYi_3QY,6
+oikan-0.0.1.10.dist-info/RECORD,,