jeremylinux / pytorch-radial-basis-function-layer Goto Github PK

Hi i'm glad i found this RBF implementation. thanks for writing it. Here's my linear model prediction so far. how do start using yours?

class Net(torch.nn.Module):
def init(self, n_feature, n_hidden, n_output):
super(Net, self).init()
self.hidden = torch.nn.Linear(n_feature, n_hidden) # hidden layer
self.predict = torch.nn.Linear(n_hidden, n_output) # output layer

def forward(self, x):
    x = F.relu(self.hidden(x))      # activation function for hidden layer
    x = self.predict(x)             # linear output
    return x

Hi ! I'm new to Pytorch and I'm looking for a way to accelerate the RBF interpolation. Has this been made available to compute on a GPU ?

Thanks

Hello, I wonder if there is a way how to check the model accuracy .... Thank you for this amazing code it really motivates me to visualize the radial basis functions.

This is should be ||x - c|| / sigma, right?