Hugging Face's logo

Spaces:
a-ragab-h-m
/
vrp-shanghai-transformer
Runtime error

App Files Community
a-ragab-h-m commited on
Commit
06a0f75
·
verified ·
1 Parent(s): 7b29e90

Update dataloader.py

Browse files
Files changed (1) hide show
  1. dataloader.py +1 -1
dataloader.py CHANGED
@@ -16,7 +16,7 @@ class VRP_Dataset(Dataset):
16
 
17
  # Load CSV data
18
  #raw_data = pd.read_csv(dataset_path)
19
- raw_data = pd.read_csv(dataset_path, nrows=15000)
20
  if len(raw_data) < dataset_size * num_nodes:
21
  raise ValueError("Not enough rows in CSV to build required dataset")
22
 
 
16
 
17
  # Load CSV data
18
  #raw_data = pd.read_csv(dataset_path)
19
+ raw_data = pd.read_csv(dataset_path, nrows=6000)
20
  if len(raw_data) < dataset_size * num_nodes:
21
  raise ValueError("Not enough rows in CSV to build required dataset")
22