Gordon Li commited on
Commit
7877659
·
1 Parent(s): 268818c

update dataset

Browse files
Files changed (1) hide show
  1. TrafficSpot.py +1 -1
TrafficSpot.py CHANGED
@@ -132,7 +132,7 @@ class TrafficSpotManager:
132
  def load_limited_traffic_spots(self, limit=10):
133
  """Load only a very limited set of traffic spots initially"""
134
  try:
135
- dataset = load_dataset("slliac/traffic-analysis", split="train")
136
  dataset_list = [row for row in dataset]
137
  dataset_list.sort(key=lambda x: x['capture_time'], reverse=True)
138
 
 
132
  def load_limited_traffic_spots(self, limit=10):
133
  """Load only a very limited set of traffic spots initially"""
134
  try:
135
+ dataset = load_dataset("slliac/isom5240-td-application-traffic-analysis", split="application")
136
  dataset_list = [row for row in dataset]
137
  dataset_list.sort(key=lambda x: x['capture_time'], reverse=True)
138