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 d252e1174b1ba554833ebde97fff6d25f3d28310 into 74ae077eafb245fa3bdca0288854b6d51f97fe60

Browse Source
This commit is contained in:
Jason Zhu 2023-01-11 15:06:16 +09:00 committed by GitHub
commit 93431bb388
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
knn/knn.go
View File

@ -7,7 +7,6 @@ import (
"errors" "errors"
"fmt" "fmt"
"github.com/gonum/matrix"
"github.com/sjwhitworth/golearn/base" "github.com/sjwhitworth/golearn/base"
"github.com/sjwhitworth/golearn/kdtree" "github.com/sjwhitworth/golearn/kdtree"
"github.com/sjwhitworth/golearn/metrics/pairwise" "github.com/sjwhitworth/golearn/metrics/pairwise"
@ -430,7 +429,7 @@ func NewKnnRegressor(distfunc string) *KNNRegressor {
func (KNN *KNNRegressor) Fit(values []float64, numbers []float64, rows int, cols int) { func (KNN *KNNRegressor) Fit(values []float64, numbers []float64, rows int, cols int) {
if rows != len(values) { if rows != len(values) {
panic(matrix.ErrShape) panic(mat.ErrShape)
} }
KNN.Data = mat.NewDense(rows, cols, numbers) KNN.Data = mat.NewDense(rows, cols, numbers)

2
linear_models/linear_regression.go
View File

@ -6,7 +6,7 @@ import (
"github.com/sjwhitworth/golearn/base" "github.com/sjwhitworth/golearn/base"
"fmt" "fmt"
_ "github.com/gonum/blas" _ "gonum.org/v1/gonum/blas"
"gonum.org/v1/gonum/mat" "gonum.org/v1/gonum/mat"
) )

3
metrics/pairwise/chebyshev.go
View File

@ -3,7 +3,6 @@ package pairwise
import ( import (
"math" "math"
"github.com/gonum/matrix"
"gonum.org/v1/gonum/mat" "gonum.org/v1/gonum/mat"
) )
@ -17,7 +16,7 @@ func (c *Chebyshev) Distance(vectorX *mat.Dense, vectorY *mat.Dense) float64 {
r1, c1 := vectorX.Dims() r1, c1 := vectorX.Dims()
r2, c2 := vectorY.Dims() r2, c2 := vectorY.Dims()
if r1 != r2 || c1 != c2 { if r1 != r2 || c1 != c2 {
panic(matrix.ErrShape) panic(mat.ErrShape)
} }
max := float64(0) max := float64(0)

3
metrics/pairwise/cranberra.go
View File

@ -3,7 +3,6 @@ package pairwise
import ( import (
"math" "math"
"github.com/gonum/matrix"
"gonum.org/v1/gonum/mat" "gonum.org/v1/gonum/mat"
) )
@ -24,7 +23,7 @@ func (c *Cranberra) Distance(vectorX *mat.Dense, vectorY *mat.Dense) float64 {
r1, c1 := vectorX.Dims() r1, c1 := vectorX.Dims()
r2, c2 := vectorY.Dims() r2, c2 := vectorY.Dims()
if r1 != r2 || c1 != c2 { if r1 != r2 || c1 != c2 {
panic(matrix.ErrShape) panic(mat.ErrShape)
} }
sum := .0 sum := .0

3
metrics/pairwise/manhattan.go
View File

@ -3,7 +3,6 @@ package pairwise
import ( import (
"math" "math"
"github.com/gonum/matrix"
"gonum.org/v1/gonum/mat" "gonum.org/v1/gonum/mat"
) )
@ -19,7 +18,7 @@ func (m *Manhattan) Distance(vectorX *mat.Dense, vectorY *mat.Dense) float64 {
r1, c1 := vectorX.Dims() r1, c1 := vectorX.Dims()
r2, c2 := vectorY.Dims() r2, c2 := vectorY.Dims()
if r1 != r2 || c1 != c2 { if r1 != r2 || c1 != c2 {
panic(matrix.ErrShape) panic(mat.ErrShape)
} }
result := .0 result := .0