andr3/fyp
1
0
Fork 0
Code Issues 29 Pull Requests Packages Projects 1 Releases Wiki Activity

fix the retrain model to eager closes #78

Browse Source
This commit is contained in:
Andre Henriques 2024-04-14 16:51:15 +01:00
parent 5f11126acd
commit 095a37189d
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
logic/models/train/train.go
View File

@ -350,7 +350,7 @@ func generateCvsExpandExp(c *Context, run_path string, model_id string, offset i
return return
} }
if file_path == "id://" { if file_path == "id://" {
f.Write([]byte(id + "," + strconv.Itoa(-1) + "\n")) f.Write([]byte(id + "," + strconv.Itoa(-2) + "\n"))
} else { } else {
return count, errors.New("TODO generateCvs to file_path " + file_path) return count, errors.New("TODO generateCvs to file_path " + file_path)
} }

2
views/py/python_model_template_expand.py
View File

@ -49,7 +49,7 @@ def pathToLabel(path):
num = table.lookup(tf.strings.as_string([path])) num = table.lookup(tf.strings.as_string([path]))
return tf.cond( return tf.cond(
tf.math.equal(num, tf.constant(-1)), tf.math.equal(num, tf.constant(-2)),
lambda: tf.zeros([depth]), lambda: tf.zeros([depth]),
lambda: tf.one_hot(table.lookup(tf.strings.as_string([path])) - diff, depth)[0] lambda: tf.one_hot(table.lookup(tf.strings.as_string([path])) - diff, depth)[0]
) )