Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,70 @@
+
+# CNN 기반 음악 감정 분류기 🎵
+
+## 개요
+
+이 모델은 음악이 전달하는 감정을 Convolutional Neural Network (CNN) 아키텍처를 사용하여 분류합니다. 
+모델은 음악 샘플로 구성된 데이터셋에서 감정 범주(Angry, Fear, Happy, Tender, Sad) 중 하나로 라벨링된 데이터를 학습했습니다.
+
+모델은 오디오 데이터를 Mel-spectrogram으로 변환한 후, CNN 레이어를 통해 음악에서 감정 상태와 관련된 패턴을 학습합니다.
+
+### 주요 기능:
+- **입력**: Mel-spectrogram으로 변환된 5초 길이의 오디오 세그먼트
+- **출력**: 다음 감정 중 하나
+  - Angry (0)
+  - Fear (1)
+  - Happy (2)
+  - Tender (3)
+  - Sad (4)
+  
+- **아키텍처**: 3개의 컨볼루션 레이어와 완전 연결층으로 구성된 CNN 모델
+
+## 데이터셋
+
+이 모델은 두 가지 데이터셋을 사용하여 학습되었습니다:
+1. **MER_audio_taffc_dataset**: 이 데이터셋은 특정 사분면(Q1~Q4)에 따라 라벨링된 음악 파일로 구성됩니다.
+2. **OSF_IO Dataset**: 또 다른 음악 파일 및 감정 라벨로 구성된 데이터셋입니다.
+
+### 데이터 전처리:
+- 각 오디오 파일을 5초 단위로 분할
+- 분할된 오디오 세그먼트를 Mel-spectrogram으로 변환
+- 두 데이터셋을 섞어서 학습 및 검증용으로 병합
+
+## 모델 상세 정보
+
+### 아키텍처:
+- **컨볼루션 레이어**: 
+  - Conv1: 32개의 필터, 커널 크기 3x3
+  - Conv2: 64개의 필터, 커널 크기 3x3
+  - Conv3: 128개의 필터, 커널 크기 3x3
+- **풀링**: 각 컨볼루션 후 MaxPooling 레이어를 사용하여 차원을 축소
+- **드롭아웃**: 드롭아웃 정규화를 사용하여 오버피팅 방지 (확률 0.5)
+- **완전 연결층**:
+  - 첫 번째 FC 레이어: 256 유닛
+  - 출력 레이어: 5개의 유닛 (5개의 감정 범주)
+
+### 손실 함수:
+- **CrossEntropyLoss**: 다중 클래스 분류를 처리하는 데 사용
+
+### 옵티마이저:
+- **Adam**: 학습률 1e-4로 설정
+
+## 학습
+
+이 모델은 병합된 데이터셋을 사용하여 10 에폭 동안 학습되었습니다. 학습과 검증 데이터 비율은 80:20으로 나누어 모델 성능을 평가했습니다.
+
+검증 정확도를 기준으로 가장 성능이 좋은 모델 체크포인트를 선택하여 저장하였습니다.
+
+## 성능
+
+- **최고 검증 정확도**: 여러 에폭 동안 튜닝한 결과, 최고 검증 정확도를 달성했습니다.
+- 최종 모델은 다양한 음악 샘플에서 감정을 인식하는 데 뛰어난 성능을 보입니다.
+
+## 사용 방법
+
+자신의 음악 데이터를 감정 분류하는 데 이 모델을 사용하려면 Hugging Face Hub에서 직접 로드할 수 있습니다:
+
+```python
+from transformers import AutoModel
+model = AutoModel.from_pretrained("jeonghyeon97/music_emotion_classifier_4")
+```