mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-11-10 15:04:45 +00:00
jacob forgot self.scal_mult
This commit is contained in:
parent
2c00f5b26e
commit
91d727b6ad
@ -112,7 +112,7 @@ class SigmoidalRegKernelArthur():
|
|||||||
def forward(self,mtx):
|
def forward(self,mtx):
|
||||||
long_in_bias=self.in_bias.repeat([1,mtx.size()[1]])
|
long_in_bias=self.in_bias.repeat([1,mtx.size()[1]])
|
||||||
long_out_bias=self.out_bias.repeat([1,mtx.size()[1]])
|
long_out_bias=self.out_bias.repeat([1,mtx.size()[1]])
|
||||||
return (scal_mult*self.sigmoid(torch.matmul(self.weights,mtx)+long_in_bias))+long_out_bias
|
return (self.scal_mult*self.sigmoid(torch.matmul(self.weights,mtx)+long_in_bias))+long_out_bias
|
||||||
|
|
||||||
class LogRegKernel():
|
class LogRegKernel():
|
||||||
parameters= []
|
parameters= []
|
||||||
@ -129,7 +129,7 @@ class LogRegKernel():
|
|||||||
def forward(self,mtx):
|
def forward(self,mtx):
|
||||||
long_in_bias=self.in_bias.repeat([1,mtx.size()[1]])
|
long_in_bias=self.in_bias.repeat([1,mtx.size()[1]])
|
||||||
long_out_bias=self.out_bias.repeat([1,mtx.size()[1]])
|
long_out_bias=self.out_bias.repeat([1,mtx.size()[1]])
|
||||||
return (scal_mult*torch.log(torch.matmul(self.weights,mtx)+long_in_bias))+long_out_bias
|
return (self.scal_mult*torch.log(torch.matmul(self.weights,mtx)+long_in_bias))+long_out_bias
|
||||||
|
|
||||||
class ExpRegKernel():
|
class ExpRegKernel():
|
||||||
parameters= []
|
parameters= []
|
||||||
@ -146,7 +146,7 @@ class ExpRegKernel():
|
|||||||
def forward(self,mtx):
|
def forward(self,mtx):
|
||||||
long_in_bias=self.in_bias.repeat([1,mtx.size()[1]])
|
long_in_bias=self.in_bias.repeat([1,mtx.size()[1]])
|
||||||
long_out_bias=self.out_bias.repeat([1,mtx.size()[1]])
|
long_out_bias=self.out_bias.repeat([1,mtx.size()[1]])
|
||||||
return (scal_mult*torch.exp(torch.matmul(self.weights,mtx)+long_in_bias))+long_out_bias
|
return (self.scal_mult*torch.exp(torch.matmul(self.weights,mtx)+long_in_bias))+long_out_bias
|
||||||
|
|
||||||
class PolyRegKernel():
|
class PolyRegKernel():
|
||||||
parameters= []
|
parameters= []
|
||||||
|
Loading…
Reference in New Issue
Block a user