muslehal/xLSTMTime

how to run main.py

mintisan opened this issue · 2 comments

How to preprocess data, set the data directory, and training commands and testing commands.

First you need to download the dataset
From here
https://drive.google.com/drive/mobile/folders/1ZOYpTUa82_jCcxIdTmyr0LXQfvaM9vIy

After that setup path to your dataset in this file datautils.py

the train and the test you can do it with this line in main file
parser.add_argument('--is_train', type=int, default=1, help='training the model')

I forgot tell you
1 mean train model
0 mean test model

parser.add_argument('--is_train', type=int, default=1, help='training the model')