search

tongnie / ImputeFormer

[KDD 2024] "ImputeFormer: Low Rankness-Induced Transformers for Generalizable Spatiotemporal Imputation"
https://arxiv.org/abs/2312.01728
MIT License
16 stars 0 forks source link

ERROR in running example #2

Open IkeYang opened 1 month ago

IkeYang commented 1 month ago

I followed the guide step by step to configure the environment and run run_traffic_imputation.py. Here is my error:

Traceback (most recent call last): File "/home/ike/YANGLuoxiao/Data/ImputeFormer/experiments/run_traffic_imputation.py", line 23, in from imputeformer.baselines import SAITS, TransformerModel, BRITS, SPINModel File "/home/ike/YANGLuoxiao/Data/ImputeFormer/experiments/imputeformer/baselines/init.py", line 3, in from .transformer import TransformerModel File "/home/ike/YANGLuoxiao/Data/ImputeFormer/experiments/imputeformer/baselines/transformer.py", line 13, in from tsl.nn.layers.graph_convs.dense_graph_conv import DenseGraphConvOrderK ModuleNotFoundError: No module named 'tsl.nn.layers.graph_convs.dense_graph_conv'

Process finished with exit code 1

tongnie commented 1 month ago

Hi Yang, our repository is still under construction yet and I will update the final version in the next month. You can get access to our model implementation now by PyPOTS, in which we upload Imputeformer as a built-in model in this open-source project.