closes #41

logic/models/train/train.go

@@ -79,7 +79,7 @@ func generateCvs(c *Context, run_path string, model_id string) (count int, err e
 		return
 	}
 
-	data, err := c.Db.Query("select mdp.id, mc.class_order, mdp.file_path from model_data_point as mdp inner join model_classes as mc on mc.id = mdp.class_id where mc.model_id = $1;", model_id)
+	data, err := c.Db.Query("select mdp.id, mc.class_order, mdp.file_path from model_data_point as mdp inner join model_classes as mc on mc.id = mdp.class_id where mc.model_id = $1 and mdp.model_mode=$2;", model_id, model_classes.DATA_POINT_MODE_TRAINING)
 	if err != nil {
 		return
 	}