30train_LSTM.py

import os
import tensorflow as tf

from tools import general
from tools.integration import train_val

if __name__ == '__main__':
    opts = general.load_config()

    method = "LSTM"
    distance_int = 3
    data_name = 'kyoto7'
    calculation_unit = "0"

    # 修订论文新增加的参数
    data_max_lenght = 300  # 数据长度, 需要探索

    dict_config_cus = {

        'model_name': opts[method]["model_name"],
        'optimizer': opts[method]["optimizer"],
        'distance_int': distance_int,
        'data_name': data_name,
        'calculation_unit': calculation_unit,

        'epochs': opts[method]["epochs"],

        'identifier': opts["public"]["identifier"],
        'purpose': opts["public"]["purpose"],
        # 'datasetsNames': ['cairo', 'milan', 'kyoto7', 'kyoto8', 'kyoto11'],
    }
    train_val(dict_config_cus, opts)