46, 50라인: FieldAwareFactorizationMachineModel에서 FeaturesLinear와 FieldAwareFactorizationMachine에 중복으로 존재 하던, 입력 데이터 x에 필드의 시작 인덱스를 더해주는 작업을 한 번에 처리하도록 코드를 수정했습니다.
24라인: torch.nn.Embedding에 입력 차원으로 들어가는 sum(field_dims)을 따로 self.input_dim이라는 하나의 변수로 선언해서 사용했습니다.
Overview
Change Log
-
To Reviewer
Issue Tags