인공지능 모델링 함수 만들기

def build_model(optimizer):

  #모델링

  model = Sequential()

  model.add( Dense(units = 6 , activation = 'relu' , input_shape=(11,) )    )

  model.add(Dense(units = 8 , activation = 'relu'  ))

  model.add( Dense(units=1 , activation = 'sigmoid'  ))

  model.compile(optimizer = optimizer, loss = 'binary_crossentropy' ,metrics= [ 'accuracy' ]   )

  return model

컴파일에 옵티마이저 파라미터에

저위에있는 빌드_모델(옵티마이저)에 괄호안에있는 옵티마이저 라는 변수를 옵티마이저(파라미터) 값으로 설정

이렇게 할경우

가능