Skip to content

Training

Mantra Patil edited this page Nov 24, 2025 · 2 revisions

Training the Model

Train the machine learning model using the extracted features.

python -m src.train

Generates:

  • models/baseline.joblib → trained RandomForest model
  • data/processed.parquet → processed feature dataset
  • Prints accuracy, precision, recall, and F1-score (synthetic dataset ≈ 0.95)

Tips

  • Retrain if you update the dataset.
  • Enable the screenshot feature for additional image hash-based feature (requires Selenium).

Clone this wiki locally