logreg js and examples

regr/main.go

@@ -15,6 +15,7 @@ func InitRegrExports(this js.Value, args []js.Value) interface{} {
 	exports.Set("ElasticNet", js.FuncOf(src.NewElasticNetJS))
 	exports.Set("Lasso", js.FuncOf(src.NewLassoJS))
 	exports.Set("R2Score", js.FuncOf(src.R2ScoreJS))
+	exports.Set("Logistic", js.FuncOf(src.NewLogisticRegressionJS))
 	return exports
 }
 

regr/src/LogisticRegressionJS.go

@@ -0,0 +1,40 @@
+//go:build js && wasm
+// +build js,wasm
+
+package src
+
+import (
+	"syscall/js"
+)
+
+func NewLogisticRegressionJS(this js.Value, args []js.Value) interface{} {
+	var (
+		epochs       = args[0].Get("epochs").Int()
+		learningRate = args[0].Get("learningRate").Float()
+	)
+	reg := &MCLogisticRegression{
+		Epochs:       epochs,
+		LearningRate: learningRate,
+	}
+	loss := 1.0
+	obj := js.Global().Get("Object").New()
+	obj.Set("fit", js.FuncOf(func(this js.Value, args []js.Value) interface{} {
+		X := JSFloatArray2D(args[0])
+		Y := JSFloatArray2D(args[1])
+		XDense := Array2DToDense(X)
+		YDense := Array2DToDense(Y)
+		reg.Fit(XDense, YDense)
+		return nil
+	}))
+	obj.Set("predict", js.FuncOf(func(this js.Value, args []js.Value) interface{} {
+		X := JSFloatArray2D(args[0])
+		XDense := Array2DToDense(X)
+		Y := reg.Predict(XDense)
+		loss = reg.Loss(XDense, Y)
+		return MatrixToJSArray2D(Y)
+	}))
+	obj.Set("loss", js.FuncOf(func(this js.Value, args []js.Value) interface{} {
+		return loss
+	}))
+	return obj
+}

regr/src/utils_js.go

@@ -5,6 +5,8 @@ package src
 
 import (
 	"syscall/js"
+
+	"gonum.org/v1/gonum/mat"
 )
 
 func JSFloatArray(arg js.Value) []float64 {
@@ -35,3 +37,26 @@ func ToJSArray[T any](arr []T) []interface{} {
 	}
 	return jsArr
 }
+
+func MatrixToJSArray(m mat.Matrix) []interface{} {
+	r, c := m.Dims()
+	data := make([]float64, r*c)
+	for i := 0; i < r; i++ {
+		for j := 0; j < c; j++ {
+			data[i*c+j] = m.At(i, j)
+		}
+	}
+	return ToJSArray(data)
+}
+
+func MatrixToJSArray2D(m mat.Matrix) []interface{} {
+	r, c := m.Dims()
+	data := make([][]interface{}, r)
+	for i := 0; i < r; i++ {
+		data[i] = make([]interface{}, c)
+		for j := 0; j < c; j++ {
+			data[i][j] = m.At(i, j)
+		}
+	}
+	return ToJSArray(data)
+}