1 모델의 정의

     

    ‘폐암 수술 환자의 생존율 예측하기’의 딥러닝 코드를 다시 한 번 옮겨 보면 다음과 같습니다.1

    실습 | 폐암 수술 환자의 생존율 예측하기

    # 텐서플로 라이브러리 안에 있는 케라스 API에서 필요한 함수들을 불러옵니다.
    from tensorflow.keras.models import Sequential
    from tensorflow.keras.layers import Dense
    
    # 데이터를 다루는 데 필요한 라이브러리를 불러옵니다.
    import numpy as np
    
    # 깃허브에 준비된 데이터를 가져옵니다.
    !git clone https://github.com/taehojo/data.git   
    
    # 준비된 수술 환자 데이터를 불러옵니다.
    Data_set = np.loadtxt("./data/ThoraricSurgery3.csv", delimiter=",") 
    
    X = Data_set[:,0:16]  # 환자의 진찰 기록을 X로 지정합니다.
    y = Data_set[:,16]    # 수술 1년 후 사망/생존 여부를 y로 지정합니다.
    
    # 딥러닝 모델의 구조를 결정합니다.
    model = Sequential()
    model.add(Dense(30, input_dim=16, activation='relu'))
    model.add(Dense(1, activation='sigmoid'))
    
    # 딥러닝 모델을 실행합니다.
    model.compile(loss='binary_crossentropy', optimizer='adam',
    metrics=['accuracy'])
    history = model.fit(X, y, epochs=5, batch_size=16)
    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.