复现STGCN
Monk丶
编程小白
展开
-
STGCN复现第四弹:解读data_utils.py
data_gen()函数:def data_gen(file_path, data_config, n_route, n_frame=21, day_slot=288): ''' Source file load and dataset generation. :param file_path: str, the file path of data source. :param data_config: tuple, the configs of dataset in tr原创 2021-04-26 09:07:05 · 1074 阅读 · 4 评论 -
STGCN复现第三弹:解读math_graph.py
scaled_laplacian函数:def scaled_laplacian(W): ''' Normalized graph Laplacian function. 标准化拉普拉斯矩阵 :param W: np.ndarray, [n_route, n_route], weighted adjacency matrix of G. :return: np.matrix, [n_route, n_route]. ''' # d -> diago..原创 2021-04-26 09:06:36 · 1361 阅读 · 2 评论 -
STGCN复现第二弹:解读main.py
import osos.environ["CUDA_VISIBLE_DEVICES"] = "0" #设置当前使用的GPU设备仅为0号设备 设备名称为'/gpu:0'from os.path import join as pjoinimport tensorflow as tftf.reset_default_graph() #函数用于清除默认图形堆栈并重置全局默认图形。#动态申请显存config = tf.ConfigProto() #tf.ConfigProto()主要的作用是配置原创 2021-04-25 22:16:22 · 1764 阅读 · 3 评论 -
STGCN复现第一弹:Anaconda 配置 Tensorflow 1.9
复现STGCN第一弹——Anaconda 配置 Tensorflow 1.91.conda常用的命令1)查看安装了哪些包conda list2)查看当前存在哪些虚拟环境conda env list conda info -e3)检查更新当前condaconda update conda2.Python创建虚拟环境conda create -n tensorflow1.9 python=3.6anaconda命令创建python版本为x.x,名字为your_env_name的虚原创 2021-04-25 16:16:45 · 849 阅读 · 0 评论