From b55fc02ede2bed5fbb705944ee227b159747aa1f Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Sat, 31 Aug 2024 21:14:31 +0200
Subject: [PATCH] switched to 50% Bert instead of 25%

---
 src/model/VeriMindSmall/config.json | 25 -------------------------
 src/model/train.py                  |  2 +-
 2 files changed, 1 insertion(+), 26 deletions(-)
 delete mode 100644 src/model/VeriMindSmall/config.json

diff --git a/src/model/VeriMindSmall/config.json b/src/model/VeriMindSmall/config.json
deleted file mode 100644
index ddc6ed2..0000000
--- a/src/model/VeriMindSmall/config.json
+++ /dev/null
@@ -1,25 +0,0 @@
-{
-  "architectures": [
-    "BertForSequenceClassification"
-  ],
-  "attention_probs_dropout_prob": 0.1,
-  "classifier_dropout": null,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.1,
-  "hidden_size": 384,
-  "initializer_range": 0.02,
-  "intermediate_size": 1536,
-  "layer_norm_eps": 1e-12,
-  "max_position_embeddings": 512,
-  "model_type": "bert",
-  "num_attention_heads": 6,
-  "num_hidden_layers": 6,
-  "pad_token_id": 0,
-  "position_embedding_type": "absolute",
-  "problem_type": "single_label_classification",
-  "torch_dtype": "float32",
-  "transformers_version": "4.44.0",
-  "type_vocab_size": 2,
-  "use_cache": true,
-  "vocab_size": 119547
-}
diff --git a/src/model/train.py b/src/model/train.py
index 730dcd9..7dd9411 100644
--- a/src/model/train.py
+++ b/src/model/train.py
@@ -114,7 +114,7 @@ if __name__ == '__main__':
     train_df, val_df = train_test_split(df, test_size=0.35, random_state=42)
 
     # Initialize and train the model
-    trainer = FakeNewsModelTrainer(size_factor=0.25)
+    trainer = FakeNewsModelTrainer(size_factor=0.5)
     train_data = trainer.prepare_data(train_df)
     val_data = trainer.prepare_data(val_df)
     trainer.train(train_data, val_data)