Added outline of dataset abstraction

DreamingRaven · Jul 14, 2022 · d909512 · d909512
1 parent 1bed91b
commit d909512
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/dataset/dataset.go b/dataset/dataset.go
@@ -0,0 +1,21 @@
+package dataset
+
+import (
+	"fmt"
+)
+
+type dataset struct {
+}
+
+type Dataset interface {
+}
+
+// NewNeuralNet initialises a new neural network
+func NewExampleDataset() Dataset {
+	return &dataset{}
+}
+
+func (ds *dataset) train() error {
+	fmt.Println("Tr")
+	return nil
+}
diff --git a/dataset/dataset_test.go b/dataset/dataset_test.go
@@ -0,0 +1,11 @@
+package dataset
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestDatasetInit(t *testing.T) {
+	ds := NewExampleDataset()
+	fmt.Println(ds)
+}