INNER CODE UNIT · Python

timeframe_0

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

    timeframe_0 = Config.timeframe_0  # таймфрейм для обучения нейросети - вход - для картинок
    draw_size = Config.draw_size  # размер стороны квадратной картинки

    # ================================================================================================================

    cur_run_folder = os.path.abspath(os.getcwd())  # текущий каталог
    data_dir = os.path.join(os.path.join(cur_run_folder, "NN"), f"training_dataset_{timeframe_0}")  # каталог с данными
    num_classes = 2  # всего классов
    epochs = 40  # Количество эпох
    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'))

View source record →