diff --git a/kan/KANLayer.py b/kan/KANLayer.py index fb35d1f1..2d461f69 100644 --- a/kan/KANLayer.py +++ b/kan/KANLayer.py @@ -28,7 +28,7 @@ class KANLayer(nn.Module): scale_base: 1D torch.float magnitude of the residual function b(x) scale_sp: 1D torch.float - mangitude of the spline function spline(x) + magnitude of the spline function spline(x) base_fun: fun residual function b(x) mask: 1D torch.float diff --git a/kan/spline.py b/kan/spline.py index 48e22508..e0c81e22 100644 --- a/kan/spline.py +++ b/kan/spline.py @@ -21,7 +21,7 @@ def B_batch(x, grid, k=0, extend=True, device='cpu'): Returns: -------- spline values : 3D torch.tensor - shape (number of splines, number of B-spline bases (coeffcients), number of samples). The numbef of B-spline bases = number of grid points + k - 1. + shape (number of splines, number of B-spline bases (coefficients), number of samples). The number of B-spline bases = number of grid points + k - 1. Example ------- @@ -119,7 +119,7 @@ def curve2coef(x_eval, y_eval, grid, k, device="cpu"): k : int the piecewise polynomial order of splines. device : str - devicde + device Example -------