|
18 | 18 |
|
19 | 19 |
|
20 | 20 | # %% |
21 | | -def generate_pairs(picks, events, stations, max_pair_dist=10, max_neighbors=50, fname="pairs.txt"): |
| 21 | +def generate_pairs(picks, events, stations, max_pair_dist=15, max_neighbors=50, fname="pairs.txt"): |
22 | 22 | ncpu = min(32, mp.cpu_count()) |
23 | 23 | neigh = NearestNeighbors(radius=max_pair_dist, n_neighbors=max_neighbors, n_jobs=ncpu) |
24 | 24 | neigh.fit(events[["x_km", "y_km", "z_km"]].values) |
@@ -92,18 +92,19 @@ def generate_pairs(picks, events, stations, max_pair_dist=10, max_neighbors=50, |
92 | 92 | # with open(args.config, "r") as fp: |
93 | 93 | # config = json.load(fp) |
94 | 94 | config.update(vars(args)) |
95 | | - print(json.dumps(config, indent=4, sort_keys=True)) |
96 | 95 |
|
97 | 96 | # %% CCTorch |
98 | | - # config["max_pair_dist_km"] = 10 |
99 | | - # config["max_neighbors"] = 50 |
100 | | - # config["max_obs"] = 100 |
101 | | - # config["min_obs"] = 8 |
| 97 | + config["max_pair_dist_km"] = 15 |
| 98 | + config["max_neighbors"] = 50 |
| 99 | + config["max_obs"] = 100 |
| 100 | + config["min_obs"] = 8 |
102 | 101 | # config["nt"] = 400 |
103 | 102 | # config["maxdepth"] = 60 |
104 | 103 | # config["mindepth"] = 0 |
105 | 104 | # config["max_epicenter_dist_km"] = 200.0 |
106 | 105 |
|
| 106 | + print(json.dumps(config, indent=4, sort_keys=True)) |
| 107 | + |
107 | 108 | # %% |
108 | 109 | def plot_templates(templates, events, picks): |
109 | 110 | templates = templates - np.nanmean(templates, axis=(-1), keepdims=True) |
|
0 commit comments