uppdated training script
Browse files- finance/train_finbert.py +1 -4
finance/train_finbert.py
CHANGED
|
@@ -105,10 +105,7 @@ def main():
|
|
| 105 |
print(f"Training complete. Saving final model to '{output_dir}'...")
|
| 106 |
trainer.save_model(output_dir)
|
| 107 |
tokenizer.save_pretrained(output_dir)
|
| 108 |
-
print("
|
| 109 |
-
print("SUCCESS: Model saved to your Google Drive in the 'Colab_Data' folder.")
|
| 110 |
-
print("You can now download the 'finbert_emotion_model' folder and place it in your project's 'finance/' directory.")
|
| 111 |
-
print("-----" * 10)
|
| 112 |
|
| 113 |
if __name__ == "__main__":
|
| 114 |
main()
|
|
|
|
| 105 |
print(f"Training complete. Saving final model to '{output_dir}'...")
|
| 106 |
trainer.save_model(output_dir)
|
| 107 |
tokenizer.save_pretrained(output_dir)
|
| 108 |
+
print("Model and tokenizer saved successfully.")
|
|
|
|
|
|
|
|
|
|
| 109 |
|
| 110 |
if __name__ == "__main__":
|
| 111 |
main()
|