jrsimuix
/

issue1038

Model card Files Files and versions Community

jrsimuix commited on Nov 19

Commit

c0a7224

•

1 Parent(s): e1ff6a2

export labels

Files changed (1) hide show

nodejs/generate_model.js +1 -0

nodejs/generate_model.js CHANGED Viewed

@@ -55,6 +55,7 @@ async function trainModel() {
     await model.fit(images, labels, { epochs: 5 });
     await model.save('file://./saved-model');  // Save in TensorFlow.js format
     console.log('Model saved as TensorFlow.js format');
 }

     await model.fit(images, labels, { epochs: 5 });
     await model.save('file://./saved-model');  // Save in TensorFlow.js format
+	fs.writeFileSync('./labelMap.json', JSON.stringify(dataset.labelMap));
     console.log('Model saved as TensorFlow.js format');
 }