[MS AI School] 1차 프로젝트 Record - 2
다른 팀원분은 GRU 모델을 사용하여 모델 학습 후 백엔드 진행을 맡았고,나는 Two-Tower 모델을 구현을 해보기로 하였다.각각의 모델로 구현 후, 코드 리뷰를 해보면서 기능을 테스트 해보기로 했다.그래서 오늘은 내가 이틀 동안 구현해 본 추천 시스템을 위한 Two-Tower 모델 학습 파이프라인을 설명해보려고 한다.데이터 전처리 # --- 1. 데이터 로드 및 전처리 --- print("\n--- 1. Loading and Preprocessing Data ---") parquet_files = [f"{data_folder_path}/{i:012d}.parquet" for i in range(4)] print(f"Attempting to load {len(parquet_fi..
2025.06.11