Spaces:

a-ragab-h-m
/

vrp-shanghai-transformer

Running

a-ragab-h-m commited on Jun 5

Commit

2860c78

verified ·

1 Parent(s): eec7179

Update dataloader.py

Files changed (1) hide show

dataloader.py CHANGED Viewed

@@ -28,7 +28,7 @@ class VRP_Dataset(Dataset):
             print(f"✅ Dataset downloaded to temporary path: {dataset_path}")
         # تحميل البيانات من CSV محلي أو بعد التنزيل
-        raw_data = pd.read_csv(dataset_path, nrows=150000)
         if len(raw_data) < dataset_size * num_nodes:
             raise ValueError("Not enough rows in CSV to build required dataset")

             print(f"✅ Dataset downloaded to temporary path: {dataset_path}")
         # تحميل البيانات من CSV محلي أو بعد التنزيل
+        raw_data = pd.read_csv(dataset_path, nrows=6000)
         if len(raw_data) < dataset_size * num_nodes:
             raise ValueError("Not enough rows in CSV to build required dataset")