OrgLion-ML/golearn
1
0
Fork 0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-28 13:48:56 +08:00
Code Issues Releases Wiki Activity

Merge pull request #147 from Sentimentron/issue-146

Browse Source 
metrics: update dot-product to match new mat64 version
This commit is contained in:
Richard Townsend 2016-09-28 15:23:48 +01:00 committed by GitHub
commit 9345938082
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
metrics/pairwise/euclidean.go
View File

@ -14,7 +14,9 @@ func NewEuclidean() *Euclidean {
// InnerProduct computes a Eucledian inner product.
func (e *Euclidean) InnerProduct(vectorX *mat64.Dense, vectorY *mat64.Dense) float64 {
result := mat64.Dot(vectorX, vectorY)
subVector := mat64.NewDense(0, 0, nil)
subVector.MulElem(vectorX, vectorY)
result := mat64.Sum(subVector)
return result
}

4
metrics/pairwise/poly_kernel.go
View File

@ -18,7 +18,9 @@ func NewPolyKernel(degree int) *PolyKernel {
// InnerProduct computes the inner product through a kernel trick
// K(x, y) = (x^T y + 1)^d
func (p *PolyKernel) InnerProduct(vectorX *mat64.Dense, vectorY *mat64.Dense) float64 {
result := mat64.Dot(vectorX, vectorY)
subVectorX := vectorX.ColView(0)
subVectorY := vectorY.ColView(0)
result := mat64.Dot(subVectorX, subVectorY)
result = math.Pow(result+1, float64(p.degree))
return result