앞에서 정의한 훈련과 레이블 데이터셋을 정규화하여 분포를 조정합니다.
코드 7-67 훈련과 테스트 데이터셋 정규화
ms = MinMaxScaler()
ss = StandardScaler()
X_ss = ss.fit_transform(X)
y_ms = ms.fit_transform(y)
X_train = X_ss[:200, :]
X_test = X_ss[200:, :]
y_train = y_ms[:200, :]
y_test = y_ms[200:, :]
print("Training Shape", X_train.shape, y_train.shape)
print("Testing Shape", X_test.shape, y_test.shape)
다음은 훈련과 테스트 데이터셋의 형태에 대한 출력 결과입니다.
Training Shape (200, 5) (200, 1) Testing Shape (53, 5) (53, 1)