Commit cd5eded4 authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

fixing dataset position

parent efa0b119
Loading
Loading
Loading
Loading
+4 −3
Original line number Diff line number Diff line
@@ -38,9 +38,10 @@ def load_kaggle(shard_size=10000, num_shards_per_batch=4,
  time1 = time.time()
  ############################################################## TIMING
  # Set some global variables up top
  train_files = ("./KAGGLE_training_disguised_combined_full.csv.gz")
  valid_files = ("./KAGGLE_test1_disguised_combined_full.csv.gz")
  test_files = ("./KAGGLE_test2_disguised_combined_full.csv.gz")
  current_dir = os.path.dirname(os.path.realpath(__file__))
  train_files = os.path.join(current_dir,"KAGGLE_training_disguised_combined_full.csv.gz")
  valid_files = os.path.join(current_dir,"KAGGLE_test1_disguised_combined_full.csv.gz")
  test_files = os.path.join(current_dir,"KAGGLE_test2_disguised_combined_full.csv.gz")

  # Featurize KAGGLE dataset
  print("About to featurize KAGGLE dataset.")