4 모델 실행하기

모델을 정의하고 컴파일하고 나면 이제 실행시킬 차례입니다. 앞서 컴파일 단계에서 정해진 환경을 주어진 데이터를 불러 실행시킬 때 사용되는 함수는 다음과 같이 model.fit() 부분입니다.

history = model.fit(X, y, epochs=5, batch_size=16)

이 부분을 설명하기에 앞서 용어를 다시 한 번 정리해 보겠습니다.

주어진 폐암 수술 환자의 수술 후 생존 여부 데이터는 총 470명의 환자에게서 16개의 정보를 정리한 것입니다. 이때 각 정보를 ‘속성’이라고 합니다. 그리고 생존 여부를 클래스, 가로 한 줄에 해당하는 각 환자의 정보를 각각 ‘샘플’이라고 합니다. 주어진 데이터에는 총 470개의 샘플이 각각 16개씩의 속성을 가지고 있는 것이라고 앞서 설명한 바 있습니다.

그림 10-2 | 폐암 환자 생존율 예측 데이터의 샘플, 속성, 클래스 구분

TIP