commit 64b9f77b9e03f58782caf018d54ae9f0f76cc714 parent 294d1ffc3d30c1c2f842eec9c9c32f1048a386ba Author: MTRNord <mtrnord1@gmail.com> Date: Wed, 28 Sep 2022 01:20:33 +0200 Remove wandb. Newer model is seemingly worse than the previous one Diffstat:
5 files changed, 15 insertions(+), 8 deletions(-)
diff --git a/model_v2.py b/model_v2.py @@ -12,9 +12,6 @@ from nltk.corpus import stopwords from tensorflow import keras from tensorflow.keras.preprocessing.sequence import pad_sequences from tensorflow.keras.preprocessing.text import Tokenizer -from wandb.keras import WandbCallback - -import wandb vocab_size = 1000 embedding_dim = 16 @@ -198,7 +195,7 @@ def train_hyperparamters(data, labels_final, tuner): stop_early = tf.keras.callbacks.EarlyStopping( monitor='val_loss', patience=5) tuner.search(data, labels_final, epochs=5, verbose=1, validation_split=0.3, - callbacks=[hypertuner_tensorboard_callback, stop_early, progress_bar, WandbCallback()]) + callbacks=[hypertuner_tensorboard_callback, stop_early, progress_bar]) # Get the optimal hyperparameters best_hps = tuner.get_best_hyperparameters(num_trials=1)[0] @@ -220,7 +217,7 @@ def train_model(data, labels_final, best_hps, tuner): epochs=num_epochs, verbose=1, validation_split=0.3, callbacks=[tensorboard_callback, - progress_bar, WandbCallback()],) + progress_bar],) val_acc_per_epoch = history.history['val_accuracy'] best_epoch = val_acc_per_epoch.index(max(val_acc_per_epoch)) + 5 print('Best epoch: %d' % (best_epoch,)) @@ -232,7 +229,7 @@ def train_model(data, labels_final, best_hps, tuner): epochs=best_epoch, validation_split=0.3, callbacks=[hypermodel_tensorboard_callback, tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_prefix, - save_weights_only=True), progress_bar, WandbCallback(), + save_weights_only=True), progress_bar # es_callback ] ) @@ -285,8 +282,6 @@ def test_model(vectorize_layer, model): def main(): print("TensorFlow version:", tf.__version__) - # wandb.tensorboard.patch(root_logdir="logs/scalars/") - wandb.init(project="matrix-spam", entity="mtrnord") print("[Step 1/6] Loading data") vectorize_layer, data, labels_final = load_data() model = SpamDectionHyperModel( diff --git a/models/spam_keras_1664320657.7486734/keras_metadata.pb b/models/spam_keras_1664320657.7486734/keras_metadata.pb @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7cf07efbc63d36dae0013e8053f0224e16a9b866f95b2ea7253af185f2a9053a +size 7962 diff --git a/models/spam_keras_1664320657.7486734/saved_model.pb b/models/spam_keras_1664320657.7486734/saved_model.pb @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:030a13f01ae38a0faafde05312bd35b2477d8009d55f76ec9e8ef84074b1e54b +size 279350 diff --git a/models/spam_keras_1664320657.7486734/variables/variables.data-00000-of-00001 b/models/spam_keras_1664320657.7486734/variables/variables.data-00000-of-00001 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f8bdd0e0d3f7ed1ad9e36783f6c4069d43d80a4caf309dd11aea40a561d4c52 +size 255667 diff --git a/models/spam_keras_1664320657.7486734/variables/variables.index b/models/spam_keras_1664320657.7486734/variables/variables.index @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dc6845fbc3ee0e9af1e8b0d8ec4b049c51ae35214b571c08e9929124fb5a067f +size 1733