You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
360 B
18 lines
360 B
import torch
|
|
|
|
DATA_PATH = "train_and_test_hibrit_dataset_v0.csv"
|
|
MODEL_SAVE_PATH = "best_gat_model_last.pth"
|
|
|
|
TIME_WINDOW = 10
|
|
TEST_SIZE = 0.3
|
|
|
|
HIDDEN_CHANNELS = 32
|
|
HEADS = 2
|
|
DROPOUT = 0.2
|
|
LEARNING_RATE = 0.005
|
|
WEIGHT_DECAY = 1e-4
|
|
EPOCHS = 40
|
|
BATCH_SIZE = 16
|
|
|
|
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
print("Using device:", DEVICE) |