gregorkrzmanc commited on
Commit
84d750e
·
1 Parent(s): 96e40d3
Files changed (2) hide show
  1. README.md +0 -12
  2. src/model_wrapper_gradio.py +1 -1
README.md CHANGED
@@ -1,15 +1,3 @@
1
-
2
- ---
3
- title: "Jet clustering"
4
- emoji: ⚛️
5
- colorFrom: "red"
6
- colorTo: "blue"
7
- app_file: app.py
8
- sdk: docker
9
- pinned: false
10
- ---
11
-
12
-
13
  # SVJ clustering
14
  The repo has evolved from [here](https://github.com/selvaggi/mlpf) - mainly, we use the dataloader and code for reading the root files for the previous MLPF project. The preprocessing part is not really needed but it does help with performance when we are doing a lot of experiments with the same dataset.
15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # SVJ clustering
2
  The repo has evolved from [here](https://github.com/selvaggi/mlpf) - mainly, we use the dataloader and code for reading the root files for the previous MLPF project. The preprocessing part is not really needed but it does help with performance when we are doing a lot of experiments with the same dataset.
3
 
src/model_wrapper_gradio.py CHANGED
@@ -97,7 +97,7 @@ def inference(loss_str, train_dataset_str, input_text, input_text_quarks):
97
  if "GP" in loss_str:
98
  n_soft = 500
99
  if n_soft > 0:
100
- pfcands = EventDataset.pfcands_add_soft_particles(pfcands, n_soft, random_generator=np.random)
101
  event = Event(pfcands=pfcands)
102
  event_batch = concat_events([event])
103
  batch, _ = get_batch(event_batch, batch_config, torch.zeros(len(pfcands)), test=True)
 
97
  if "GP" in loss_str:
98
  n_soft = 500
99
  if n_soft > 0:
100
+ pfcands = EventDataset.pfcands_add_soft_particles(pfcands, n_soft, random_generator=np.random.RandomState(seed=0))
101
  event = Event(pfcands=pfcands)
102
  event_batch = concat_events([event])
103
  batch, _ = get_batch(event_batch, batch_config, torch.zeros(len(pfcands)), test=True)