Update TSLANet_classification.py

This commit is contained in:
Emadeldeen Eldele 2024-04-18 16:37:22 +08:00 committed by GitHub
parent 9d191631f9
commit be1b18e506
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -388,7 +388,7 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--model_id', type=str, default='TEST') parser.add_argument('--model_id', type=str, default='TEST')
parser.add_argument('--data_path', type=str, default=r'C:\Emad\datasets\UCR_2015\UCR_2015_preprocessed\OSULeaf') parser.add_argument('--data_path', type=str, default=r'data/hhar')
# Training parameters: # Training parameters:
parser.add_argument('--num_epochs', type=int, default=100) parser.add_argument('--num_epochs', type=int, default=100)