trainModel.py 729 B

12345678910111213141516171819202122232425262728
  1. import sys
  2. import os
  3. import logging
  4. from src.data_processor import DataProcessor
  5. from src.model_trainer import ModelTrainer
  6. from src.model_runner import ModelRunner
  7. # 设置日志
  8. logging.basicConfig(level=logging.INFO)
  9. logger = logging.getLogger(__name__)
  10. def main():
  11. try:
  12. # 数据集生成与处理
  13. logger.info("Generating and processing data...")
  14. data_processor = DataProcessor()
  15. data_processor.generate_raw_data()
  16. data_processor.process_data()
  17. # 模型训练
  18. logger.info("Training model...")
  19. trainer = ModelTrainer()
  20. trainer.train()
  21. except Exception as e:
  22. logger.error(f"An error occurred: {e}")
  23. if __name__ == "__main__":
  24. main()