diff options
author | Slendi <slendi@socopon.com> | 2023-11-05 01:49:19 +0200 |
---|---|---|
committer | Slendi <slendi@socopon.com> | 2023-11-05 01:52:21 +0200 |
commit | 00d3d9218d7cbabf3b8bb45417ad76b70772375f (patch) | |
tree | bd94d86472e8b84f70921efd8c3959409e753c84 | |
parent | b57e8d00f6ff73d5548b46973e96c3bc4ea3be02 (diff) |
Signed-off-by: Slendi <slendi@socopon.com>
-rwxr-xr-x | interactive.py | 8 | ||||
-rwxr-xr-x | train.py | 2 |
2 files changed, 8 insertions, 2 deletions
diff --git a/interactive.py b/interactive.py index bb06fb7..9d29c86 100755 --- a/interactive.py +++ b/interactive.py @@ -13,4 +13,10 @@ with tf.keras.utils.CustomObjectScope({'text_standardizer': text_standardizer}): model.summary() while True: - print(model.predict([input('> ')])) + inp = [i.strip() for i in input('> ').split('.')] + while "" in inp: + inp.remove("") + output = model.predict(inp) + score = sum(output) / len(output) + print(["UwU" if i > 0.5 else "Normal" for i in output]) + print("Final judgement:", "UwU" if score > 0.5 else "Normal") @@ -101,7 +101,7 @@ model.compile(loss=losses.BinaryCrossentropy(from_logits=True), optimizer='adam', metrics=tf.metrics.BinaryAccuracy(threshold=0.0)) -epochs = 10 +epochs = 12 history = model.fit( train_ds, validation_data=val_ds, |