splitting up code into seperate modules

alg/alg.go

@@ -0,0 +1,5 @@
+package alg
+
+type Alg interface {
+	Train()
+}

nn/main.go

@@ -0,0 +1,11 @@
+package nn
+
+import "github.com/Jensen-holm/ml-from-scratch/alg"
+
+type NN struct {
+	alg.Alg
+}
+
+func New(rp *RequestPayload) {
+
+}

nn/split.go

@@ -0,0 +1,5 @@
+package nn
+
+// implement train test split function
+
+func trainTestSplit() {}

nn/subset.go

@@ -0,0 +1,6 @@
+package nn
+
+// subset the data frame into just the
+// features and target that the user specify
+
+func subset() {}

request/request.go

@@ -0,0 +1,31 @@
+package request
+
+import (
+	"fmt"
+
+	"github.com/go-gota/gota/dataframe"
+	"github.com/gofiber/fiber/v2"
+)
+
+type Payload struct {
+	CSVData        string   `json:"csv_data"`
+	Features       []string `json:"features"`
+	Target         string   `json:"target"`
+	Epochs         int      `json:"epochs"`
+	HiddenSize     int      `json:"hidden_size"`
+	LearningRate   float64  `json:"learning_rate"`
+	ActivationFunc string   `json:"activation_func"`
+
+	Df dataframe.DataFrame
+}
+
+func (p *Payload) SetDf(df dataframe.DataFrame) {
+	p.Df = df
+}
+
+func NewPayload(dest *Payload, c *fiber.Ctx) error {
+	if err := c.BodyParser(dest); err != nil {
+		return fmt.Errorf("invalid JSON data")
+	}
+	return nil
+}

server.go

@@ -2,35 +2,30 @@ package main
 
 import (
 	"fmt"
+	"strings"
 
+	"github.com/Jensen-holm/ml-from-scratch/request"
+	"github.com/go-gota/gota/dataframe"
 	"github.com/gofiber/fiber/v2"
 )
 
-type RequestPayload struct {
-	CSVData        string   `json:"csv_data"`
-	Features       []string `json:"features"`
-	Target         string   `json:"target"`
-	Epochs         int      `json:"epochs"`
-	HiddenSize     int      `json:"hidden_size"`
-	LearningRate   float64  `json:"learning_rate"`
-	ActivationFunc string   `json:"activation_func"`
-}
-
 func main() {
 	app := fiber.New()
 
 	app.Post("/", func(c *fiber.Ctx) error {
+		r := new(request.Payload)
 
-		requestData := new(RequestPayload)
-
-		// parse json request data into requestData struct
-		if err := c.BodyParser(requestData); err != nil {
+		err := request.NewPayload(r, c)
+		if err != nil {
 			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
 				"error": "Invalid JSON data",
 			})
 		}
 
-		fmt.Println(requestData)
+		df := dataframe.ReadCSV(strings.NewReader(r.CSVData))
+		r.SetDf(df)
+
+		fmt.Println(r.Df)
 
 		return c.SendString("No error")
 	})