Update gen_adj_mx.py

liyaguang · Apr 24, 2020 · 3d7e4f9 · 3d7e4f9
1 parent 8780bab
commit 3d7e4f9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/gen_adj_mx.py b/scripts/gen_adj_mx.py
@@ -57,6 +57,7 @@ def get_adjacency_matrix(distance_df, sensor_ids, normalized_k=0.1):
  with open(args.sensor_ids_filename) as f:
  sensor_ids = f.read().strip().split(',')
  distance_df = pd.read_csv(args.distances_filename, dtype={'from': 'str', 'to': 'str'})
+ normalized_k = args.normalized_k
  _, sensor_id_to_ind, adj_mx = get_adjacency_matrix(distance_df, sensor_ids, normalized_k)
  # Save to pickle file.
  with open(args.output_pkl_filename, 'wb') as f: