본문 바로가기

ml_framework

cnn_layer

    # cnn 층
    def cnn_layer(self, filter_size, filter_count, padding = "same", stride = 1):
        cnn_dict = dict()

        cnn_dict['layer_type'] = "cnn"
        cnn_dict['filter_size'] = filter_size
        cnn_dict['filter_count'] = filter_count
        cnn_dict['padding'] = padding
        cnn_dict['stride'] = stride

        return cnn_dict

필터 사이즈를 입력받으면서 정사각 형태라고 가정,

 

'ml_framework' 카테고리의 다른 글

pooling  (0) 2023.04.28
padding  (0) 2023.04.28
cnn  (0) 2023.04.27
fit(delta_weight_update)  (0) 2023.04.25
fit(delta)  (0) 2023.04.24