더북(TheBook)

코드 9-29 데이터셋 로딩 및 모델 훈련

df = pd.read_csv('../chap9/data/covtype.csv')
x = df[df.columns[:54]]
y = df.Cover_Type  ------ 정답(레이블)을 Cover_Type 칼럼으로 지정

x_train, x_test, y_train, y_test = train_test_split(x, y, train_size=0.7,
                                                    random_state=90) ------ 훈련과 검증 데이터셋으로 분리하며, 전체 데이터셋 중 70%를 훈련용으로 사용

model = tf.keras.Sequential([
    tf.keras.layers.Dense(64, activation='relu',
                          input_shape=(x_train.shape[1],)),
    tf.keras.layers.Dense(64, activation='relu'),
    tf.keras.layers.Dense(8, activation='softmax')
]) ------ 출력층은 소프트맥스 활성 함수 사용

model.compile(optimizer=tf.keras.optimizers.Adam(0.001),
              loss='sparse-categorical crossentropy',
              metrics=['accuracy'])------ y가 다중 분류가 가능한 값을 갖기 때문에 sparse-categorical crossentropy 손실 함수 사용
history1 = model.fit(
    x_train, y_train,
    epochs=26, batch_size=60,
    validation_data=(x_test, y_test))  ------ 모델 훈련
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.