INNER CODE UNIT · Python

batch_size

WISEPLAT/Hackathon-Finam-NN-Trade-Robot · 3_train_neural_network.py:57

    batch_size = 10  # Размер мини-выборки
    img_height, img_width = draw_size, draw_size  # размер картинок
    input_shape = (img_height, img_width, 3)  # размерность картинки

    # # Первый тип модели
    # model = Sequential()
    # model.add(Rescaling(1. / 255))
    # model.add(Conv2D(64, (3, 3), input_shape=input_shape))
    # model.add(Activation('relu'))
    # model.add(MaxPooling2D(pool_size=(2, 2)))
    # model.add(BatchNormalization())
    # model.add(Conv2D(32, (3, 3)))
    # model.add(Activation('relu'))
    # model.add(MaxPooling2D(pool_size=(2, 2)))
    # model.add(BatchNormalization())
    # model.add(Flatten())
    # model.add(Dense(128))
    # model.add(Activation('relu'))

View source record →