2016-11-17 5 views
2

Keras를 사용하여 모델링 작업을하고 있는데 인덱스 또는 이름으로 특정 레이어를 제거 할 수 있습니까? 현재 나는 model.pop()이 작업을 수행 할 수 있지만 가장 최근에 추가 된 레이어 만 제거한다는 것을 알고 있습니다. 또한 layerstensorvariable 유형이며 numpy array 또는 list에서 수행 할 수있는 특정 요소를 제거하는 방법을 알지 못합니다. BTW 나는 Theano 백엔드를 사용하고 있습니다.Kera가 model.fit() 이후에 레이어를 제거합니다.

+0

'layers'와'tensorvariable'에 관한 질문의 두 번째 부분에 대해 자세히 설명 할 수 있습니까? 미리 감사드립니다. – petezurich

답변

1

model.pop()은 단지 마지막으로 추가 한 레이어를 제거하고 중간 레이어를 삭제하는 다른 문서화 된 방법은 없다는 것이 맞습니다.

당신은 항상 같은 중간 계층의 출력을 얻을 수 있습니다 :

base_model = VGG19(weights='imagenet') 
model = Model(inputs=base_model.input, outputs=base_model.get_layer('block4_pool').output) 

예는 여기에서 촬영 : 그 위에 새 레이어를 추가보다 https://keras.io/applications/

합니다.