feat: closes #23

logic/models/run.go

@@ -7,7 +7,6 @@ import (
 	"net/http"
 	"os"
 	"path"
-	"strconv"
 
 	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
 	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
@@ -131,6 +130,8 @@ func handleRun(handle *Handle) {
         vi := 0
         var predictions = results[0].Value().([][]float32)[0]
 
+        fmt.Println(predictions)
+
         for i, v := range predictions {
             if v > vmax {
                 vi = i
@@ -140,9 +141,24 @@ func handleRun(handle *Handle) {
 
         os.RemoveAll(run_path)
 
+        rows, err := handle.Db.Query("select name from model_classes where model_id=$1 and class_order=$2;", model.Id, vi)
+        if err != nil { return Error500(err) }
+        if !rows.Next() {
+            LoadDefineTemplate(w, "/models/edit.html", "run-model-card", c.AddMap(AnyMap{
+                "Model": model,
+                "NotFound": true,
+                "Result": nil,
+            }))
+            return nil
+        }
+
+        var name string
+        if err = rows.Scan(&name); err != nil { return nil }
+
+
         LoadDefineTemplate(w, "/models/edit.html", "run-model-card", c.AddMap(AnyMap{
             "Model": model,
-            "Result": strconv.Itoa(vi),
+            "Result": name,
         }))
 		return nil
 	})

logic/models/train/train.go

@@ -83,12 +83,6 @@ func generateCvs(handle *Handle, run_path string, model_id string)  (count int,
     if err != nil { return }
     defer data.Close()
 
-    type row struct {
-        path string
-        class_order int
-    }
-
-
     f, err := os.Create(path.Join(run_path, "train.csv"))
     if err != nil { return }
     defer f.Close()