linpershey commited on
Commit
6023585
1 Parent(s): 7c9c311

add finer grained classes

Browse files
Files changed (1) hide show
  1. sheet.py +7 -3
sheet.py CHANGED
@@ -566,7 +566,7 @@ def main(args):
566
  """
567
 
568
  ## 讀取資料名單 ##
569
- data = get_leads(args.data_path).head(20)
570
 
571
  ## 進行爬蟲與分析 ##
572
  # crawled_results = crawl_results(data)
@@ -629,7 +629,10 @@ category2supercategory = {
629
  "韓式料理(含火鍋,烤肉)": "中式",
630
  "西餐廳(含美式,義式,墨式)": "西式",
631
  "中式": "中式",
632
- "西式": "西式"
 
 
 
633
  }
634
 
635
  supercategory2category = {
@@ -644,7 +647,8 @@ supercategory2category = {
644
  "燒烤",
645
  "韓式料理(含火鍋,烤肉)"
646
  ],
647
- "西式": ["西餐廳(含美式,義式,墨式)"]
 
648
  }
649
 
650
  if __name__=='__main__':
 
566
  """
567
 
568
  ## 讀取資料名單 ##
569
+ data = get_leads(args.data_path).tail(20)
570
 
571
  ## 進行爬蟲與分析 ##
572
  # crawled_results = crawl_results(data)
 
629
  "韓式料理(含火鍋,烤肉)": "中式",
630
  "西餐廳(含美式,義式,墨式)": "西式",
631
  "中式": "中式",
632
+ "西式": "西式",
633
+ "西餐廳(餐酒館、酒吧、標吧、pub、lounge bar)": "西式",
634
+ "西餐廳(土耳其、漢堡、薯條、法式、歐式、印度)": "西式",
635
+ "早餐": ""
636
  }
637
 
638
  supercategory2category = {
 
647
  "燒烤",
648
  "韓式料理(含火鍋,烤肉)"
649
  ],
650
+ "西式": ["西餐廳(含美式,義式,墨式)", "西餐廳(餐酒館、酒吧、標吧、pub、lounge bar)", "西餐廳(土耳其、漢堡、法式、歐式、印度)"],
651
+ "": ["早餐"]
652
  }
653
 
654
  if __name__=='__main__':