yotitinogs
yotitinogs
"My results are not efficient with unbalanced data."
--------------------------------------------------------------------------- AssertionError Traceback (most recent call last) Cell In[26], [line 10](vscode-notebook-cell:?execution_count=26&line=10) [7](vscode-notebook-cell:?execution_count=26&line=7) device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') [8](vscode-notebook-cell:?execution_count=26&line=8) print(device) ---> [10](vscode-notebook-cell:?execution_count=26&line=10) model = KAN(width=[15,6,3,1], grid=5, k=3,device=device) #, grid_range=(0,1))...
model(dataset['train_input']) model.plot(beta=10) --------------------------------------------------------------------------- RuntimeError Traceback (most recent call last) Cell In[51], [line 1](vscode-notebook-cell:?execution_count=51&line=1) ----> [1](vscode-notebook-cell:?execution_count=51&line=1) model(dataset['train_input'].double()) [2](vscode-notebook-cell:?execution_count=51&line=2) model.plot(beta=10) File c:\Users\thnog\AppData\Local\Programs\Python\Python311\Lib\site-packages\torch\nn\modules\module.py:1532, in Module._wrapped_call_impl(self, *args, **kwargs) [1530](file:///C:/Users/thnog/AppData/Local/Programs/Python/Python311/Lib/site-packages/torch/nn/modules/module.py:1530) return self._compiled_call_impl(*args, **kwargs) # type:...