From 6507b05d00cc260c4cabe6841142c1e6ad04d8f4 Mon Sep 17 00:00:00 2001 From: jcollopy-tulane Date: Mon, 29 Apr 2024 12:44:47 -0500 Subject: [PATCH] ] --- nlp/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nlp/cli.py b/nlp/cli.py index 44c629a..b7052ce 100644 --- a/nlp/cli.py +++ b/nlp/cli.py @@ -234,7 +234,7 @@ def tokenize(sentences, max_length=100, padding='max_length'): model.compile( optimizer=optimizer, loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True), - metrics = ["accuracy") + metrics = ["accuracy"]) model.fit( x=train_dataset,