하이퍼 파라미터(모델링할 때 사용자가 직접 설정해 주는 값)를 설정한다. embedding_dim은 임베딩할 벡터의 차원을 나타낸다.

embedding_dim = 64

클래스의 수는 분류될 예측값의 종류다. 정답값이 원-핫 인코딩으로 되어 있기 때문에 정답값의 열의 수가 예측값의 종류가 된다.

n_class = y_train.shape[1]
n_class

실행 결과

3