Basic AI Project

이 프로젝트는 PyTorch를 사용하여 MNIST 데이터셋에 대한 간단한 MLP 분류 모델을 구현합니다.

프로젝트 구조

• data/: MNIST 데이터셋이 저장되는 디렉토리
• models/: 모델 정의 파일
• notebooks/: 데이터 탐색을 위한 Jupyter 노트북
• utils/: 데이터 로딩 유틸리티
• main.py: 메인 실행 스크립트

실행 방법

1. 필요한 패키지 설치:

   pip install torch torchvision

2. 모델 학습 및 평가:

   python main.py

참고

• 이 프로젝트는 교육 목적으로 제공되며, 각 팀원이 역할에 따라 코드를 수정하고 기능을 추가할 수 있습니다.