changes to prompt, hyperparameters

This commit is contained in:
Umar Javed
2025-10-28 12:01:47 +05:00
parent 42b2db426e
commit 512821511f

View File

@@ -396,13 +396,12 @@ def main():
print("\nCheck the generated chart for detailed RMSLE metrics!") print("\nCheck the generated chart for detailed RMSLE metrics!")
print("\nPrice prediction fine-tuning process completed!") print("\nPrice prediction fine-tuning process completed!")
print("\nImproved configuration to prevent overfitting:")
print(" Uses pickle files (train.pkl, test.pkl)") print(" Uses pickle files (train.pkl, test.pkl)")
print(" 750 training examples, 100 validation examples") print(" 750 training examples, 100 validation examples")
print(" 1 epoch to prevent overfitting") print(" 1 epoch")
print(" Learning rate: 0.5 (higher for better generalization)") print(" Learning rate: 0.5")
print(" Batch size: 8 (larger for stability)") print(" Batch size: 8")
print(" Assertive system prompt (forces predictions)") print(" Assertive system prompt")
print(" Proper RMSLE evaluation using Tester class") print(" Proper RMSLE evaluation using Tester class")
print(" Weights and Biases integration") print(" Weights and Biases integration")