feat: closes #38

logic/models/classes/list.go

@@ -48,7 +48,7 @@ func HandleList(handle *Handle) {
             return Error500(nil)
         }
 
-        rows, err := handle.Db.Query("select id, file_path, model_mode from model_data_point where class_id=$1 limit 10 offset $2;", id, page * 10)
+        rows, err := handle.Db.Query("select id, file_path, model_mode, status from model_data_point where class_id=$1 limit 10 offset $2;", id, page * 10)
         if err != nil {
             return Error500(err)
         }
@@ -58,13 +58,14 @@ func HandleList(handle *Handle) {
             Id string
             FilePath string
             Mode int
+            Status int
         }
 
         got := []baserow{}
 
         for rows.Next() {
             nrow := baserow{}
-            err = rows.Scan(&nrow.Id, &nrow.FilePath, &nrow.Mode)
+            err = rows.Scan(&nrow.Id, &nrow.FilePath, &nrow.Mode, &nrow.Status)
             if err != nil {
                 return Error500(err)
             }

logic/models/train/train.go

@@ -326,7 +326,7 @@ func trainModel(c *Context, model *BaseModel) {
 }
 
 func removeFailedDataPoints(db *sql.DB, model *BaseModel) (err error) {
-	rows, err := db.Query("select id from model_data_point as mdp join model_classes as mc on mc.id=mpd.class_id where mc.model_id=$1 and mdp.status=-1;", model.Id)
+	rows, err := db.Query("select mdp.id from model_data_point as mdp join model_classes as mc on mc.id=mdp.class_id where mc.model_id=$1 and mdp.status=-1;", model.Id)
 	if err != nil {
 		return
 	}
@@ -346,6 +346,7 @@ func removeFailedDataPoints(db *sql.DB, model *BaseModel) (err error) {
         }
 	}
 
+    _, err = db.Exec("delete from model_data_point as mdp using model_classes as mc where mdp.class_id = mc.id and mc.model_id=$1 and mdp.status=-1;", model.Id)
 	return
 }
 
@@ -418,7 +419,7 @@ func handleTrain(handle *Handle) {
 			return c.Error500(err)
 		}
 
-		err = removeFailedDataPoints(c.Db, model.Id)
+		err = removeFailedDataPoints(c.Db, model)
 		if err != nil {
 			return c.Error500(err)
 		}