diff --git a/examples/atlas/sweep_results/Blood_ans.csv b/examples/atlas/sweep_results/Blood_ans.csv
index 51c2d46d..f878ef5e 100644
--- a/examples/atlas/sweep_results/Blood_ans.csv
+++ b/examples/atlas/sweep_results/Blood_ans.csv
@@ -376,7 +376,7 @@ a5d95a42-0137-496f-8a60-101e17f263c8,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.44528302550315857,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/l3suyx48|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/h55dfb6n,https://wandb.ai/xzy11632/dance-dev/sweeps/rcl4mr3y,https://wandb.ai/xzy11632/dance-dev/sweeps/nc90i5yd","- type: filter.gene
+",0.4452830255031585,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/l3suyx48|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/h55dfb6n,https://wandb.ai/xzy11632/dance-dev/sweeps/rcl4mr3y,https://wandb.ai/xzy11632/dance-dev/sweeps/nc90i5yd","- type: filter.gene
   target: FilterGenesPercentile
   params:
     min_val: 6
@@ -477,7 +477,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.19449999928474423,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/67nc05qo|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/968h8p8g,https://wandb.ai/xzy11632/dance-dev/sweeps/fuzr2xth,https://wandb.ai/xzy11632/dance-dev/sweeps/u6dg202m","- type: filter.gene
+",0.1944999992847442,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/67nc05qo|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/968h8p8g,https://wandb.ai/xzy11632/dance-dev/sweeps/fuzr2xth,https://wandb.ai/xzy11632/dance-dev/sweeps/u6dg202m","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -526,7 +526,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.20600000023841855,all_total_runs:510|all_finished_runs:415|step2_total_runs:450|step2_finished_runs:357|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:81.37%,all_total_runs:210|all_finished_runs:163|step2_total_runs:150|step2_finished_runs:108|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:17|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:77.62%,all_total_runs:150|all_finished_runs:106|step2_total_runs:90|step2_finished_runs:52|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:15|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:70.67%,all_total_runs:600|all_finished_runs:429|step2_total_runs:540|step2_finished_runs:370|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:71.50%,False,False,False,False
+",0.2060000002384185,all_total_runs:510|all_finished_runs:415|step2_total_runs:450|step2_finished_runs:357|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:81.37%,all_total_runs:210|all_finished_runs:163|step2_total_runs:150|step2_finished_runs:108|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:17|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:77.62%,all_total_runs:150|all_finished_runs:106|step2_total_runs:90|step2_finished_runs:52|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:15|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:70.67%,all_total_runs:600|all_finished_runs:429|step2_total_runs:540|step2_finished_runs:370|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:71.50%,False,False,False,False
 456e8b9b-f872-488b-871d-94534090a865,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/wri3w9ug|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/9c166tcq,https://wandb.ai/xzy11632/dance-dev/sweeps/0j1n56kx,https://wandb.ai/xzy11632/dance-dev/sweeps/3761sh9x","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
@@ -579,7 +579,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.23399999737739563,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/lv0fx66b|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/8ddug6xp,https://wandb.ai/xzy11632/dance-dev/sweeps/idxr1y9z,https://wandb.ai/xzy11632/dance-dev/sweeps/0iz6wobo","- type: filter.gene
+",0.2339999973773956,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/lv0fx66b|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/8ddug6xp,https://wandb.ai/xzy11632/dance-dev/sweeps/idxr1y9z,https://wandb.ai/xzy11632/dance-dev/sweeps/0iz6wobo","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -699,7 +699,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.32600000500679016,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/538iyppx|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/fj7pyf1f,https://wandb.ai/xzy11632/dance-dev/sweeps/es7cum5d,https://wandb.ai/xzy11632/dance-dev/sweeps/o5ozuhv2","- type: filter.gene
+",0.3260000050067901,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/538iyppx|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/fj7pyf1f,https://wandb.ai/xzy11632/dance-dev/sweeps/es7cum5d,https://wandb.ai/xzy11632/dance-dev/sweeps/o5ozuhv2","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -747,7 +747,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.33550000190734863,all_total_runs:510|all_finished_runs:378|step2_total_runs:450|step2_finished_runs:356|step3_0_total_runs:20|step3_0_finished_runs:1|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:74.12%,all_total_runs:210|all_finished_runs:164|step2_total_runs:150|step2_finished_runs:104|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:78.10%,all_total_runs:150|all_finished_runs:105|step2_total_runs:90|step2_finished_runs:49|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:16|finished_rate:70.00%,all_total_runs:600|all_finished_runs:448|step2_total_runs:540|step2_finished_runs:388|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:74.67%,False,False,False,False
+",0.3355000019073486,all_total_runs:510|all_finished_runs:378|step2_total_runs:450|step2_finished_runs:356|step3_0_total_runs:20|step3_0_finished_runs:1|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:74.12%,all_total_runs:210|all_finished_runs:164|step2_total_runs:150|step2_finished_runs:104|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:78.10%,all_total_runs:150|all_finished_runs:105|step2_total_runs:90|step2_finished_runs:49|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:16|finished_rate:70.00%,all_total_runs:600|all_finished_runs:448|step2_total_runs:540|step2_finished_runs:388|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:74.67%,False,False,False,False
 01209dce-3575-4bed-b1df-129f57fbc031,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/sthk65mb|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/zncxm6s3,https://wandb.ai/xzy11632/dance-dev/sweeps/3fzlad77,https://wandb.ai/xzy11632/dance-dev/sweeps/knpn4gwt","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
@@ -967,7 +967,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.22349999845027924,all_total_runs:510|all_finished_runs:401|step2_total_runs:450|step2_finished_runs:351|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:12|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:78.63%,all_total_runs:210|all_finished_runs:175|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:83.33%,all_total_runs:150|all_finished_runs:85|step2_total_runs:90|step2_finished_runs:63|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:2|finished_rate:56.67%,all_total_runs:600|all_finished_runs:487|step2_total_runs:540|step2_finished_runs:428|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:81.17%,False,False,True,False
+",0.2234999984502792,all_total_runs:510|all_finished_runs:401|step2_total_runs:450|step2_finished_runs:351|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:12|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:78.63%,all_total_runs:210|all_finished_runs:175|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:83.33%,all_total_runs:150|all_finished_runs:85|step2_total_runs:90|step2_finished_runs:63|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:2|finished_rate:56.67%,all_total_runs:600|all_finished_runs:487|step2_total_runs:540|step2_finished_runs:428|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:81.17%,False,False,True,False
 2a498ace-872a-4935-984b-1afa70fd9886,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/bq07gszd|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/bktx311f,https://wandb.ai/xzy11632/dance-dev/sweeps/5mdhsnx2,https://wandb.ai/xzy11632/dance-dev/sweeps/fcfzcmn5","- type: filter.gene
   target: FilterGenesPlaceHolder
 - type: normalize
@@ -1076,8 +1076,8 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.19499999284744263,all_total_runs:510|all_finished_runs:390|step2_total_runs:450|step2_finished_runs:350|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:76.47%,all_total_runs:210|all_finished_runs:170|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:14|finished_rate:80.95%,all_total_runs:150|all_finished_runs:93|step2_total_runs:90|step2_finished_runs:55|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:1|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:62.00%,all_total_runs:600|all_finished_runs:477|step2_total_runs:540|step2_finished_runs:427|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:13|finished_rate:79.50%,False,False,False,False
-2adb1f8a-a6b1-4909-8ee8-484814e2d4bf,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/bym68vxb|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/ecznwhs3,https://wandb.ai/xzy11632/dance-dev/sweeps/kqtogj5m,https://wandb.ai/xzy11632/dance-dev/sweeps/w9mcnr87","- type: filter.gene
+",0.1949999928474426,all_total_runs:510|all_finished_runs:390|step2_total_runs:450|step2_finished_runs:350|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:76.47%,all_total_runs:210|all_finished_runs:170|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:14|finished_rate:80.95%,all_total_runs:150|all_finished_runs:93|step2_total_runs:90|step2_finished_runs:55|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:1|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:62.00%,all_total_runs:600|all_finished_runs:477|step2_total_runs:540|step2_finished_runs:427|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:13|finished_rate:79.50%,False,False,False,False
+2adb1f8a-a6b1-4909-8ee8-484814e2d4bf,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/0rvmzkia|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/e9ndxdji,https://wandb.ai/xzy11632/dance-dev/sweeps/6lndzud2,https://wandb.ai/xzy11632/dance-dev/sweeps/kboc5b0g","- type: filter.gene
   target: FilterGenesPercentile
   params: {}
 - type: normalize
@@ -1098,7 +1098,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.2469999939203262,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/ydgm0lth|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/26noy2pa,https://wandb.ai/xzy11632/dance-dev/sweeps/2f5d1v5q,https://wandb.ai/xzy11632/dance-dev/sweeps/tzq3n1vg","- type: filter.gene
+",0.24699999392032623,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/ydgm0lth|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/26noy2pa,https://wandb.ai/xzy11632/dance-dev/sweeps/2f5d1v5q,https://wandb.ai/xzy11632/dance-dev/sweeps/tzq3n1vg","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -1173,7 +1173,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.289000004529953,all_total_runs:510|all_finished_runs:369|step2_total_runs:450|step2_finished_runs:327|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:2|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:72.35%,all_total_runs:196|all_finished_runs:151|step2_total_runs:150|step2_finished_runs:114|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:6|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:17|finished_rate:77.04%,all_total_runs:150|all_finished_runs:120|step2_total_runs:90|step2_finished_runs:60|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.00%,all_total_runs:600|all_finished_runs:497|step2_total_runs:540|step2_finished_runs:439|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:82.83%,False,False,False,False
+",0.289000004529953,all_total_runs:493|all_finished_runs:376|step2_total_runs:450|step2_finished_runs:336|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:76.27%,all_total_runs:196|all_finished_runs:151|step2_total_runs:150|step2_finished_runs:114|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:6|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:17|finished_rate:77.04%,all_total_runs:150|all_finished_runs:120|step2_total_runs:90|step2_finished_runs:60|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.00%,all_total_runs:600|all_finished_runs:497|step2_total_runs:540|step2_finished_runs:439|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:82.83%,False,False,False,False
 3faad104-2ab8-4434-816d-474d8d2641db,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/fizus47n|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/opykbimx,https://wandb.ai/xzy11632/dance-dev/sweeps/i46fxdho,https://wandb.ai/xzy11632/dance-dev/sweeps/0rpl7b1w","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
@@ -1274,7 +1274,7 @@ c7775e88-49bf-4ba2-a03b-93f00447c958,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.23849999904632568,all_total_runs:510|all_finished_runs:408|step2_total_runs:450|step2_finished_runs:350|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.00%,all_total_runs:210|all_finished_runs:158|step2_total_runs:150|step2_finished_runs:98|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:75.24%,all_total_runs:130|all_finished_runs:75|step2_total_runs:90|step2_finished_runs:52|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:3|finished_rate:57.69%,all_total_runs:600|all_finished_runs:496|step2_total_runs:540|step2_finished_runs:436|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:82.67%,False,False,True,False
+",0.2384999990463256,all_total_runs:510|all_finished_runs:408|step2_total_runs:450|step2_finished_runs:350|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.00%,all_total_runs:210|all_finished_runs:158|step2_total_runs:150|step2_finished_runs:98|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:75.24%,all_total_runs:130|all_finished_runs:75|step2_total_runs:90|step2_finished_runs:52|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:3|finished_rate:57.69%,all_total_runs:600|all_finished_runs:496|step2_total_runs:540|step2_finished_runs:436|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:82.67%,False,False,True,False
 4c4cd77c-8fee-4836-9145-16562a8782fe,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/lsbn6s01|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/zbwjy7sy,https://wandb.ai/xzy11632/dance-dev/sweeps/7qh0tjpd,https://wandb.ai/xzy11632/dance-dev/sweeps/0exovwss","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
@@ -2146,7 +2146,7 @@ d7d7e89c-c93a-422d-8958-9b4a90b69558,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.1609999984502792,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/viqji691|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/ch8dy9h8,https://wandb.ai/xzy11632/dance-dev/sweeps/dcd34yk6,https://wandb.ai/xzy11632/dance-dev/sweeps/nimckczz","- type: filter.gene
+",0.16099999845027924,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/viqji691|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/ch8dy9h8,https://wandb.ai/xzy11632/dance-dev/sweeps/dcd34yk6,https://wandb.ai/xzy11632/dance-dev/sweeps/nimckczz","- type: filter.gene
   target: FilterGenesPercentile
   params:
     min_val: 10
diff --git a/examples/atlas/sweep_results/Brain_ans.csv b/examples/atlas/sweep_results/Brain_ans.csv
index 71b0a5be..4145ef17 100644
--- a/examples/atlas/sweep_results/Brain_ans.csv
+++ b/examples/atlas/sweep_results/Brain_ans.csv
@@ -386,7 +386,7 @@ Dataset_id,cta_scdeepsort,cta_scdeepsort_best_yaml,cta_scdeepsort_best_res,cta_c
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.3549999892711639,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/hpa8itbf|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/yxhw6hrg,https://wandb.ai/xzy11632/dance-dev/sweeps/ogwvhde7,https://wandb.ai/xzy11632/dance-dev/sweeps/1mzn11g0","- type: filter.gene
+",0.35499998927116394,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/hpa8itbf|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/yxhw6hrg,https://wandb.ai/xzy11632/dance-dev/sweeps/ogwvhde7,https://wandb.ai/xzy11632/dance-dev/sweeps/1mzn11g0","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -1432,7 +1432,7 @@ Dataset_id,cta_scdeepsort,cta_scdeepsort_best_yaml,cta_scdeepsort_best_res,cta_c
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.3120000064373016,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/66gvcj96|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/dfkmrl1i,https://wandb.ai/xzy11632/dance-dev/sweeps/ciietvtw,https://wandb.ai/xzy11632/dance-dev/sweeps/f1mqyetb","- type: filter.gene
+",0.31200000643730164,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/66gvcj96|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/dfkmrl1i,https://wandb.ai/xzy11632/dance-dev/sweeps/ciietvtw,https://wandb.ai/xzy11632/dance-dev/sweeps/f1mqyetb","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -1497,7 +1497,7 @@ Dataset_id,cta_scdeepsort,cta_scdeepsort_best_yaml,cta_scdeepsort_best_res,cta_c
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.34150001406669617,all_total_runs:600|all_finished_runs:483|step2_total_runs:540|step2_finished_runs:423|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.50%,all_total_runs:150|all_finished_runs:104|step2_total_runs:90|step2_finished_runs:69|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:69.33%,all_total_runs:510|all_finished_runs:346|step2_total_runs:450|step2_finished_runs:343|step3_0_total_runs:20|step3_0_finished_runs:1|step3_1_total_runs:20|step3_1_finished_runs:1|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:67.84%,all_total_runs:196|all_finished_runs:144|step2_total_runs:150|step2_finished_runs:118|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:6|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:73.47%,False,False,False,False
+",0.3415000140666961,all_total_runs:600|all_finished_runs:483|step2_total_runs:540|step2_finished_runs:423|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.50%,all_total_runs:150|all_finished_runs:104|step2_total_runs:90|step2_finished_runs:69|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:69.33%,all_total_runs:510|all_finished_runs:346|step2_total_runs:450|step2_finished_runs:343|step3_0_total_runs:20|step3_0_finished_runs:1|step3_1_total_runs:20|step3_1_finished_runs:1|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:67.84%,all_total_runs:196|all_finished_runs:144|step2_total_runs:150|step2_finished_runs:118|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:6|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:73.47%,False,False,False,False
 9372df2d-13d6-4fac-980b-919a5b7eb483,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/nmy20c1k|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/bf71t9iq,https://wandb.ai/xzy11632/dance-dev/sweeps/hj5g01na,https://wandb.ai/xzy11632/dance-dev/sweeps/6yqm4q5t","- type: filter.gene
   target: FilterGenesPercentile
   params: {}
@@ -1660,7 +1660,7 @@ Dataset_id,cta_scdeepsort,cta_scdeepsort_best_yaml,cta_scdeepsort_best_res,cta_c
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.4510000050067901,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/ol3ztzsp|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/4y8mkdbc,https://wandb.ai/xzy11632/dance-dev/sweeps/227yfu9d,https://wandb.ai/xzy11632/dance-dev/sweeps/p7ic9i6t","- type: filter.gene
+",0.45100000500679016,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/ol3ztzsp|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/4y8mkdbc,https://wandb.ai/xzy11632/dance-dev/sweeps/227yfu9d,https://wandb.ai/xzy11632/dance-dev/sweeps/p7ic9i6t","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -2038,7 +2038,7 @@ e8681d74-ac9e-4be5-be14-1cf1bbd54dd7,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.4550000131130218,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/9ulcm4zs|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/lc65vlu9,https://wandb.ai/xzy11632/dance-dev/sweeps/102sw7wh,https://wandb.ai/xzy11632/dance-dev/sweeps/xg2gclj5","- type: filter.gene
+",0.45500001311302185,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/9ulcm4zs|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/lc65vlu9,https://wandb.ai/xzy11632/dance-dev/sweeps/102sw7wh,https://wandb.ai/xzy11632/dance-dev/sweeps/xg2gclj5","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -2091,7 +2091,7 @@ e8681d74-ac9e-4be5-be14-1cf1bbd54dd7,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.47200000286102295,all_total_runs:600|all_finished_runs:484|step2_total_runs:540|step2_finished_runs:425|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:80.67%,all_total_runs:150|all_finished_runs:98|step2_total_runs:90|step2_finished_runs:62|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:65.33%,all_total_runs:493|all_finished_runs:322|step2_total_runs:450|step2_finished_runs:301|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:65.31%,all_total_runs:182|all_finished_runs:77|step2_total_runs:150|step2_finished_runs:65|step3_0_total_runs:20|step3_0_finished_runs:12|step3_1_total_runs:6|step3_1_finished_runs:0|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:42.31%,True,False,False,False
+",0.4720000028610229,all_total_runs:600|all_finished_runs:484|step2_total_runs:540|step2_finished_runs:425|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:80.67%,all_total_runs:150|all_finished_runs:98|step2_total_runs:90|step2_finished_runs:62|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:65.33%,all_total_runs:493|all_finished_runs:322|step2_total_runs:450|step2_finished_runs:301|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:65.31%,all_total_runs:182|all_finished_runs:77|step2_total_runs:150|step2_finished_runs:65|step3_0_total_runs:20|step3_0_finished_runs:12|step3_1_total_runs:6|step3_1_finished_runs:0|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:42.31%,True,False,False,False
 93cb76aa-a84b-4a92-8e6c-66a914e26d4c,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/aqfmsmpx|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/sxj7ndgn,https://wandb.ai/xzy11632/dance-dev/sweeps/kf6pxhij,https://wandb.ai/xzy11632/dance-dev/sweeps/kfpas6cb","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
@@ -2784,7 +2784,7 @@ f64e1be1-de15-4d27-8da4-82225cd4c035,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.4739454090595245,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/zzzd70pb|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/h6iokqel,https://wandb.ai/xzy11632/dance-dev/sweeps/8tcdt2ii,https://wandb.ai/xzy11632/dance-dev/sweeps/g7s2t70w","- type: filter.gene
+",0.47394540905952454,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/zzzd70pb|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/h6iokqel,https://wandb.ai/xzy11632/dance-dev/sweeps/8tcdt2ii,https://wandb.ai/xzy11632/dance-dev/sweeps/g7s2t70w","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -2878,7 +2878,7 @@ ed11cc3e-2947-407c-883c-c53b043917c3,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.468804657459259,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/d2u4p3ti|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/5tarnk33,https://wandb.ai/xzy11632/dance-dev/sweeps/pyebz1b7,https://wandb.ai/xzy11632/dance-dev/sweeps/vtfdyrpk","- type: filter.gene
+",0.46880465745925903,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/d2u4p3ti|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/5tarnk33,https://wandb.ai/xzy11632/dance-dev/sweeps/pyebz1b7,https://wandb.ai/xzy11632/dance-dev/sweeps/vtfdyrpk","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -2979,7 +2979,7 @@ ed11cc3e-2947-407c-883c-c53b043917c3,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.3235000073909759,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/o748tsyg|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/py2yh3r8,https://wandb.ai/xzy11632/dance-dev/sweeps/cdz2iob0,https://wandb.ai/xzy11632/dance-dev/sweeps/ax6j7ylt","- type: filter.gene
+",0.32350000739097595,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/o748tsyg|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/py2yh3r8,https://wandb.ai/xzy11632/dance-dev/sweeps/cdz2iob0,https://wandb.ai/xzy11632/dance-dev/sweeps/ax6j7ylt","- type: filter.gene
   target: FilterGenesPercentile
   params: {}
 - type: normalize
diff --git a/examples/atlas/sweep_results/Heart_ans.csv b/examples/atlas/sweep_results/Heart_ans.csv
index 601977da..644f267b 100644
--- a/examples/atlas/sweep_results/Heart_ans.csv
+++ b/examples/atlas/sweep_results/Heart_ans.csv
@@ -715,7 +715,7 @@ c5d88abe-f23a-45fa-a534-788985e93dad,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.387499988079071,all_total_runs:210|all_finished_runs:178|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:84.76%,all_total_runs:558|all_finished_runs:349|step2_total_runs:540|step2_finished_runs:349|step3_0_total_runs:6|step3_0_finished_runs:0|step3_1_total_runs:6|step3_1_finished_runs:0|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:62.54%,all_total_runs:150|all_finished_runs:102|step2_total_runs:90|step2_finished_runs:62|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:68.00%,all_total_runs:493|all_finished_runs:360|step2_total_runs:450|step2_finished_runs:339|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:73.02%,False,False,False,False
+",0.38749998807907104,all_total_runs:210|all_finished_runs:178|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:18|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:84.76%,all_total_runs:558|all_finished_runs:349|step2_total_runs:540|step2_finished_runs:349|step3_0_total_runs:6|step3_0_finished_runs:0|step3_1_total_runs:6|step3_1_finished_runs:0|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:62.54%,all_total_runs:150|all_finished_runs:102|step2_total_runs:90|step2_finished_runs:62|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:68.00%,all_total_runs:493|all_finished_runs:360|step2_total_runs:450|step2_finished_runs:339|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:73.02%,False,False,False,False
 1009f384-b12d-448e-ba9f-1b7d2ecfbb4e,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/gwh6la6w|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/3vsbqktb,https://wandb.ai/xzy11632/dance-dev/sweeps/xt9unkek,https://wandb.ai/xzy11632/dance-dev/sweeps/kb3pvbli","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
@@ -1688,7 +1688,7 @@ f15e263b-6544-46cb-a46e-e33ab7ce8347,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.324999988079071,all_total_runs:196|all_finished_runs:154|step2_total_runs:150|step2_finished_runs:119|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:16|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:78.57%,all_total_runs:600|all_finished_runs:499|step2_total_runs:540|step2_finished_runs:442|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:17|finished_rate:83.17%,all_total_runs:150|all_finished_runs:71|step2_total_runs:90|step2_finished_runs:70|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:47.33%,all_total_runs:453|all_finished_runs:348|step2_total_runs:450|step2_finished_runs:348|step3_0_total_runs:1|step3_0_finished_runs:0|step3_1_total_runs:1|step3_1_finished_runs:0|step3_2_total_runs:1|step3_2_finished_runs:0|finished_rate:76.82%,False,False,False,True
+",0.32499998807907104,all_total_runs:196|all_finished_runs:154|step2_total_runs:150|step2_finished_runs:119|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:16|step3_2_total_runs:6|step3_2_finished_runs:0|finished_rate:78.57%,all_total_runs:600|all_finished_runs:499|step2_total_runs:540|step2_finished_runs:442|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:17|finished_rate:83.17%,all_total_runs:150|all_finished_runs:71|step2_total_runs:90|step2_finished_runs:70|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:1|finished_rate:47.33%,all_total_runs:453|all_finished_runs:348|step2_total_runs:450|step2_finished_runs:348|step3_0_total_runs:1|step3_0_finished_runs:0|step3_1_total_runs:1|step3_1_finished_runs:0|step3_2_total_runs:1|step3_2_finished_runs:0|finished_rate:76.82%,False,False,False,True
 f75f2ff4-2884-4c2d-b375-70de37a34507,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/5ueqmi7v|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/1qtlda25,https://wandb.ai/xzy11632/dance-dev/sweeps/ir5rr5hz,https://wandb.ai/xzy11632/dance-dev/sweeps/krbtx7e0","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
@@ -2105,7 +2105,7 @@ fa27492b-82ff-4ab7-ac61-0e2b184eee67,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.3454999923706054,all_total_runs:210|all_finished_runs:172|step2_total_runs:150|step2_finished_runs:118|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:15|finished_rate:81.90%,all_total_runs:600|all_finished_runs:497|step2_total_runs:540|step2_finished_runs:441|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:17|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:82.83%,all_total_runs:150|all_finished_runs:82|step2_total_runs:90|step2_finished_runs:63|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:2|step3_2_total_runs:20|step3_2_finished_runs:17|finished_rate:54.67%,all_total_runs:510|all_finished_runs:405|step2_total_runs:450|step2_finished_runs:346|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:79.41%,False,False,False,True
+",0.34549999237060547,all_total_runs:210|all_finished_runs:172|step2_total_runs:150|step2_finished_runs:118|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:15|finished_rate:81.90%,all_total_runs:600|all_finished_runs:497|step2_total_runs:540|step2_finished_runs:441|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:17|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:82.83%,all_total_runs:150|all_finished_runs:82|step2_total_runs:90|step2_finished_runs:63|step3_0_total_runs:20|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:2|step3_2_total_runs:20|step3_2_finished_runs:17|finished_rate:54.67%,all_total_runs:510|all_finished_runs:405|step2_total_runs:450|step2_finished_runs:346|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:79.41%,False,False,False,True
 f7c1c579-2dc0-47e2-ba19-8165c5a0e353,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/5o741t7i|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/63x9jmta,https://wandb.ai/xzy11632/dance-dev/sweeps/rlwzwx5n,https://wandb.ai/xzy11632/dance-dev/sweeps/jvvacpgh","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
@@ -2438,4 +2438,4 @@ ed852810-a003-4386-9846-1638362cee39,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.2975000143051147,all_total_runs:210|all_finished_runs:180|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:85.71%,all_total_runs:600|all_finished_runs:501|step2_total_runs:540|step2_finished_runs:442|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:83.50%,all_total_runs:150|all_finished_runs:135|step2_total_runs:90|step2_finished_runs:75|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:90.00%,all_total_runs:459|all_finished_runs:352|step2_total_runs:450|step2_finished_runs:352|step3_0_total_runs:3|step3_0_finished_runs:0|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:3|step3_2_finished_runs:0|finished_rate:76.69%,False,False,False,False
+",0.29750001430511475,all_total_runs:210|all_finished_runs:180|step2_total_runs:150|step2_finished_runs:120|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:85.71%,all_total_runs:600|all_finished_runs:501|step2_total_runs:540|step2_finished_runs:442|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:83.50%,all_total_runs:150|all_finished_runs:135|step2_total_runs:90|step2_finished_runs:75|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:90.00%,all_total_runs:459|all_finished_runs:352|step2_total_runs:450|step2_finished_runs:352|step3_0_total_runs:3|step3_0_finished_runs:0|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:3|step3_2_finished_runs:0|finished_rate:76.69%,False,False,False,False
diff --git a/examples/atlas/sweep_results/Intestine_ans.csv b/examples/atlas/sweep_results/Intestine_ans.csv
index 42afe5d2..214df82f 100644
--- a/examples/atlas/sweep_results/Intestine_ans.csv
+++ b/examples/atlas/sweep_results/Intestine_ans.csv
@@ -64,7 +64,7 @@ e40c6272-af77-4a10-9385-62a398884f27,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.350659430027008,all_total_runs:476|all_finished_runs:359|step2_total_runs:450|step2_finished_runs:359|step3_0_total_runs:3|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:0|step3_2_total_runs:3|step3_2_finished_runs:0|finished_rate:75.42%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/2ygbd0g1|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/wa5xd2jo,https://wandb.ai/xzy11632/dance-dev/sweeps/njbvga25,https://wandb.ai/xzy11632/dance-dev/sweeps/1ueyprxc","- type: filter.gene
+",0.35065943002700806,all_total_runs:476|all_finished_runs:359|step2_total_runs:450|step2_finished_runs:359|step3_0_total_runs:3|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:0|step3_2_total_runs:3|step3_2_finished_runs:0|finished_rate:75.42%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/2ygbd0g1|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/wa5xd2jo,https://wandb.ai/xzy11632/dance-dev/sweeps/njbvga25,https://wandb.ai/xzy11632/dance-dev/sweeps/1ueyprxc","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -187,7 +187,7 @@ e40c6272-af77-4a10-9385-62a398884f27,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.4136029481887817,all_total_runs:493|all_finished_runs:400|step2_total_runs:450|step2_finished_runs:360|step3_0_total_runs:3|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:81.14%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/hpjhyj9q|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/wyv5vmis,https://wandb.ai/xzy11632/dance-dev/sweeps/8qobi3e5,https://wandb.ai/xzy11632/dance-dev/sweeps/jkwnilze","- type: filter.gene
+",0.41360294818878174,all_total_runs:493|all_finished_runs:400|step2_total_runs:450|step2_finished_runs:360|step3_0_total_runs:3|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:81.14%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/hpjhyj9q|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/wyv5vmis,https://wandb.ai/xzy11632/dance-dev/sweeps/8qobi3e5,https://wandb.ai/xzy11632/dance-dev/sweeps/jkwnilze","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -348,7 +348,7 @@ f7c1c579-2dc0-47e2-ba19-8165c5a0e353(Intestine),"step2:https://wandb.ai/xzy11632
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.48100000619888306,all_total_runs:600|all_finished_runs:510|step2_total_runs:540|step2_finished_runs:450|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:85.00%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/kxkmzv41|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/467ux2eq,https://wandb.ai/xzy11632/dance-dev/sweeps/213zwuix,https://wandb.ai/xzy11632/dance-dev/sweeps/qa7qjg5q","- type: filter.gene
+",0.481000006198883,all_total_runs:600|all_finished_runs:510|step2_total_runs:540|step2_finished_runs:450|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:85.00%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/kxkmzv41|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/467ux2eq,https://wandb.ai/xzy11632/dance-dev/sweeps/213zwuix,https://wandb.ai/xzy11632/dance-dev/sweeps/qa7qjg5q","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -613,7 +613,7 @@ fa27492b-82ff-4ab7-ac61-0e2b184eee67(Intestine),"step2:https://wandb.ai/xzy11632
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.449999988079071,all_total_runs:510|all_finished_runs:408|step2_total_runs:450|step2_finished_runs:349|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.00%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/mvhlvdxl|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/0zk0bilm,https://wandb.ai/xzy11632/dance-dev/sweeps/z9xnldun,https://wandb.ai/xzy11632/dance-dev/sweeps/xctbkm4s","- type: filter.gene
+",0.44999998807907104,all_total_runs:510|all_finished_runs:408|step2_total_runs:450|step2_finished_runs:349|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.00%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/mvhlvdxl|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/0zk0bilm,https://wandb.ai/xzy11632/dance-dev/sweeps/z9xnldun,https://wandb.ai/xzy11632/dance-dev/sweeps/xctbkm4s","- type: filter.gene
   target: FilterGenesPlaceHolder
 - type: normalize
   target: ScaleFeature
diff --git a/examples/atlas/sweep_results/Kidney_ans.csv b/examples/atlas/sweep_results/Kidney_ans.csv
index 64af019b..89d989d0 100644
--- a/examples/atlas/sweep_results/Kidney_ans.csv
+++ b/examples/atlas/sweep_results/Kidney_ans.csv
@@ -403,7 +403,7 @@ dea717d4-7bc0-4e46-950f-fd7e1cc8df7d,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.2269999980926513,all_total_runs:510|all_finished_runs:398|step2_total_runs:450|step2_finished_runs:341|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:78.04%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/cdcv6mna|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/ne54nu6k,https://wandb.ai/xzy11632/dance-dev/sweeps/0z8zj8do,https://wandb.ai/xzy11632/dance-dev/sweeps/bi91nl8w","- type: filter.gene
+",0.22699999809265137,all_total_runs:510|all_finished_runs:398|step2_total_runs:450|step2_finished_runs:341|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:78.04%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/cdcv6mna|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/ne54nu6k,https://wandb.ai/xzy11632/dance-dev/sweeps/0z8zj8do,https://wandb.ai/xzy11632/dance-dev/sweeps/bi91nl8w","- type: filter.gene
   target: FilterGenesPlaceHolder
 - type: normalize
   target: ScaleFeature
@@ -458,7 +458,7 @@ dea717d4-7bc0-4e46-950f-fd7e1cc8df7d,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.27649998664855957,all_total_runs:600|all_finished_runs:443|step2_total_runs:540|step2_finished_runs:403|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:16|step3_2_total_runs:20|step3_2_finished_runs:7|finished_rate:73.83%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/mb2cyn4c|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/uha7jepq,https://wandb.ai/xzy11632/dance-dev/sweeps/nq1sd8jx,https://wandb.ai/xzy11632/dance-dev/sweeps/fhvkz6ju","- type: filter.gene
+",0.2764999866485595,all_total_runs:600|all_finished_runs:443|step2_total_runs:540|step2_finished_runs:403|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:16|step3_2_total_runs:20|step3_2_finished_runs:7|finished_rate:73.83%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/mb2cyn4c|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/uha7jepq,https://wandb.ai/xzy11632/dance-dev/sweeps/nq1sd8jx,https://wandb.ai/xzy11632/dance-dev/sweeps/fhvkz6ju","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -617,7 +617,7 @@ be39785b-67cb-4177-be19-a40ee3747e45,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.4289999902248382,all_total_runs:510|all_finished_runs:360|step2_total_runs:450|step2_finished_runs:338|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:1|step3_2_total_runs:20|step3_2_finished_runs:3|finished_rate:70.59%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/1a0xicia|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/wunz1pqq,https://wandb.ai/xzy11632/dance-dev/sweeps/0n5dswxy,https://wandb.ai/xzy11632/dance-dev/sweeps/13if0wtd","- type: filter.gene
+",0.42899999022483826,all_total_runs:510|all_finished_runs:360|step2_total_runs:450|step2_finished_runs:338|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:1|step3_2_total_runs:20|step3_2_finished_runs:3|finished_rate:70.59%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/1a0xicia|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/wunz1pqq,https://wandb.ai/xzy11632/dance-dev/sweeps/0n5dswxy,https://wandb.ai/xzy11632/dance-dev/sweeps/13if0wtd","- type: filter.gene
   target: FilterGenesPercentile
   params:
     min_val: 8
@@ -830,7 +830,7 @@ be39785b-67cb-4177-be19-a40ee3747e45,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.2049999982118606,all_total_runs:493|all_finished_runs:400|step2_total_runs:450|step2_finished_runs:360|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:81.14%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/sk7ynfj9|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/64bshk8k,https://wandb.ai/xzy11632/dance-dev/sweeps/y9l1vpb7,https://wandb.ai/xzy11632/dance-dev/sweeps/v6t2nb4k","- type: filter.gene
+",0.20499999821186063,all_total_runs:493|all_finished_runs:400|step2_total_runs:450|step2_finished_runs:360|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:81.14%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/sk7ynfj9|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/64bshk8k,https://wandb.ai/xzy11632/dance-dev/sweeps/y9l1vpb7,https://wandb.ai/xzy11632/dance-dev/sweeps/v6t2nb4k","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -1223,7 +1223,7 @@ f801b7a9-80a6-4d09-9161-71474deb58ae,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.33247917890548706,all_total_runs:600|all_finished_runs:483|step2_total_runs:540|step2_finished_runs:423|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.50%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/xx97kth1|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/2ij5ijlv,https://wandb.ai/xzy11632/dance-dev/sweeps/euzr7ujg,https://wandb.ai/xzy11632/dance-dev/sweeps/dlrnxq72","- type: filter.gene
+",0.332479178905487,all_total_runs:600|all_finished_runs:483|step2_total_runs:540|step2_finished_runs:423|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:80.50%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/xx97kth1|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/2ij5ijlv,https://wandb.ai/xzy11632/dance-dev/sweeps/euzr7ujg,https://wandb.ai/xzy11632/dance-dev/sweeps/dlrnxq72","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -1436,7 +1436,7 @@ fd072bc3-2dfb-46f8-b4e3-467cb3223182,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.28049999475479126,all_total_runs:600|all_finished_runs:487|step2_total_runs:540|step2_finished_runs:427|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:81.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/rfy7muvr|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/0pdtdvge,https://wandb.ai/xzy11632/dance-dev/sweeps/7qoqqm29,https://wandb.ai/xzy11632/dance-dev/sweeps/z3tsakmq","- type: filter.gene
+",0.2804999947547912,all_total_runs:600|all_finished_runs:487|step2_total_runs:540|step2_finished_runs:427|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:81.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/rfy7muvr|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/0pdtdvge,https://wandb.ai/xzy11632/dance-dev/sweeps/7qoqqm29,https://wandb.ai/xzy11632/dance-dev/sweeps/z3tsakmq","- type: filter.gene
   target: FilterGenesPercentile
   params: {}
 - type: normalize
@@ -1480,7 +1480,7 @@ fd072bc3-2dfb-46f8-b4e3-467cb3223182,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.2730000019073486,all_total_runs:493|all_finished_runs:380|step2_total_runs:450|step2_finished_runs:340|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:77.08%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/fv5txyhd|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/gftuks1f,https://wandb.ai/xzy11632/dance-dev/sweeps/vhtoxock,https://wandb.ai/xzy11632/dance-dev/sweeps/1qsjz3ai","- type: filter.gene
+",0.27300000190734863,all_total_runs:493|all_finished_runs:380|step2_total_runs:450|step2_finished_runs:340|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:77.08%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/fv5txyhd|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/gftuks1f,https://wandb.ai/xzy11632/dance-dev/sweeps/vhtoxock,https://wandb.ai/xzy11632/dance-dev/sweeps/1qsjz3ai","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -1595,7 +1595,7 @@ fd072bc3-2dfb-46f8-b4e3-467cb3223182,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.2249999940395355,all_total_runs:510|all_finished_runs:363|step2_total_runs:450|step2_finished_runs:339|step3_0_total_runs:20|step3_0_finished_runs:1|step3_1_total_runs:20|step3_1_finished_runs:3|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:71.18%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/h267r9gw|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/gsf3qy1c,https://wandb.ai/xzy11632/dance-dev/sweeps/dzembwwg,https://wandb.ai/xzy11632/dance-dev/sweeps/iygud7ig","- type: filter.gene
+",0.22499999403953552,all_total_runs:510|all_finished_runs:363|step2_total_runs:450|step2_finished_runs:339|step3_0_total_runs:20|step3_0_finished_runs:1|step3_1_total_runs:20|step3_1_finished_runs:3|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:71.18%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/h267r9gw|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/gsf3qy1c,https://wandb.ai/xzy11632/dance-dev/sweeps/dzembwwg,https://wandb.ai/xzy11632/dance-dev/sweeps/iygud7ig","- type: filter.gene
   target: FilterGenesPercentile
   params: {}
 - type: normalize
diff --git a/examples/atlas/sweep_results/Pancreas_ans.csv b/examples/atlas/sweep_results/Pancreas_ans.csv
index f42aba38..5f69bbff 100644
--- a/examples/atlas/sweep_results/Pancreas_ans.csv
+++ b/examples/atlas/sweep_results/Pancreas_ans.csv
@@ -306,7 +306,7 @@ b07e5164-baf6-43d2-bdba-5a249d0da879,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.39201876521110535,all_total_runs:600|all_finished_runs:499|step2_total_runs:540|step2_finished_runs:440|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:83.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/acvxgobi|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/367ly4lx,https://wandb.ai/xzy11632/dance-dev/sweeps/qyt8n1ye,https://wandb.ai/xzy11632/dance-dev/sweeps/yuvualyj","- type: filter.gene
+",0.3920187652111053,all_total_runs:600|all_finished_runs:499|step2_total_runs:540|step2_finished_runs:440|step3_0_total_runs:20|step3_0_finished_runs:19|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:83.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/acvxgobi|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/367ly4lx,https://wandb.ai/xzy11632/dance-dev/sweeps/qyt8n1ye,https://wandb.ai/xzy11632/dance-dev/sweeps/yuvualyj","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -331,7 +331,7 @@ b07e5164-baf6-43d2-bdba-5a249d0da879,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.3896713554859161,all_total_runs:210|all_finished_runs:129|step2_total_runs:150|step2_finished_runs:112|step3_0_total_runs:20|step3_0_finished_runs:2|step3_1_total_runs:20|step3_1_finished_runs:9|step3_2_total_runs:20|step3_2_finished_runs:6|finished_rate:61.43%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/4y7bjr73|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/2bftkn2f,https://wandb.ai/xzy11632/dance-dev/sweeps/6wksp40t,https://wandb.ai/xzy11632/dance-dev/sweeps/b0igv55h","- type: filter.gene
+",0.38967135548591614,all_total_runs:210|all_finished_runs:129|step2_total_runs:150|step2_finished_runs:112|step3_0_total_runs:20|step3_0_finished_runs:2|step3_1_total_runs:20|step3_1_finished_runs:9|step3_2_total_runs:20|step3_2_finished_runs:6|finished_rate:61.43%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/4y7bjr73|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/2bftkn2f,https://wandb.ai/xzy11632/dance-dev/sweeps/6wksp40t,https://wandb.ai/xzy11632/dance-dev/sweeps/b0igv55h","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -432,7 +432,7 @@ b07e5164-baf6-43d2-bdba-5a249d0da879,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.4769999980926513,all_total_runs:193|all_finished_runs:114|step2_total_runs:150|step2_finished_runs:89|step3_0_total_runs:20|step3_0_finished_runs:5|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:59.07%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/g85syyzg|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/cdggwljw,https://wandb.ai/xzy11632/dance-dev/sweeps/natn1hgy,https://wandb.ai/xzy11632/dance-dev/sweeps/b0eefc5r","- type: filter.gene
+",0.47699999809265137,all_total_runs:193|all_finished_runs:114|step2_total_runs:150|step2_finished_runs:89|step3_0_total_runs:20|step3_0_finished_runs:5|step3_1_total_runs:3|step3_1_finished_runs:0|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:59.07%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/g85syyzg|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/cdggwljw,https://wandb.ai/xzy11632/dance-dev/sweeps/natn1hgy,https://wandb.ai/xzy11632/dance-dev/sweeps/b0eefc5r","- type: filter.gene
   target: FilterGenesPercentile
   params: {}
 - type: normalize
@@ -507,7 +507,7 @@ ff45e623-7f5f-46e3-b47d-56be0341f66b,"step2:https://wandb.ai/xzy11632/dance-dev/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.41350001096725464,all_total_runs:600|all_finished_runs:499|step2_total_runs:540|step2_finished_runs:442|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:83.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/xrzp33gj|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/erm7pgbk,https://wandb.ai/xzy11632/dance-dev/sweeps/z5w6mxqq,https://wandb.ai/xzy11632/dance-dev/sweeps/6gwdbnnf","- type: filter.gene
+",0.4135000109672546,all_total_runs:600|all_finished_runs:499|step2_total_runs:540|step2_finished_runs:442|step3_0_total_runs:20|step3_0_finished_runs:17|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:83.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/xrzp33gj|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/erm7pgbk,https://wandb.ai/xzy11632/dance-dev/sweeps/z5w6mxqq,https://wandb.ai/xzy11632/dance-dev/sweeps/6gwdbnnf","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -648,7 +648,7 @@ ff45e623-7f5f-46e3-b47d-56be0341f66b,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.3772102296352386,all_total_runs:196|all_finished_runs:105|step2_total_runs:150|step2_finished_runs:66|step3_0_total_runs:6|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:53.57%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/p87lac44|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/lfrx75bu,https://wandb.ai/xzy11632/dance-dev/sweeps/9dafryxc,https://wandb.ai/xzy11632/dance-dev/sweeps/yh7p9cq3","- type: filter.gene
+",0.37721022963523865,all_total_runs:196|all_finished_runs:105|step2_total_runs:150|step2_finished_runs:66|step3_0_total_runs:6|step3_0_finished_runs:0|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:19|finished_rate:53.57%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/p87lac44|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/lfrx75bu,https://wandb.ai/xzy11632/dance-dev/sweeps/9dafryxc,https://wandb.ai/xzy11632/dance-dev/sweeps/yh7p9cq3","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -769,7 +769,7 @@ ff45e623-7f5f-46e3-b47d-56be0341f66b,"step2:https://wandb.ai/xzy11632/dance-dev/
   params:
     config_dict:
       label_channel: cell_type
-",0.402999997138977,all_total_runs:210|all_finished_runs:126|step2_total_runs:150|step2_finished_runs:99|step3_0_total_runs:20|step3_0_finished_runs:8|step3_1_total_runs:20|step3_1_finished_runs:10|step3_2_total_runs:20|step3_2_finished_runs:9|finished_rate:60.00%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/q7l6m2qp|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/5go16fns,https://wandb.ai/xzy11632/dance-dev/sweeps/i63ze0b1,https://wandb.ai/xzy11632/dance-dev/sweeps/e7g4eglj","- type: filter.gene
+",0.40299999713897705,all_total_runs:210|all_finished_runs:126|step2_total_runs:150|step2_finished_runs:99|step3_0_total_runs:20|step3_0_finished_runs:8|step3_1_total_runs:20|step3_1_finished_runs:10|step3_2_total_runs:20|step3_2_finished_runs:9|finished_rate:60.00%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/q7l6m2qp|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/5go16fns,https://wandb.ai/xzy11632/dance-dev/sweeps/i63ze0b1,https://wandb.ai/xzy11632/dance-dev/sweeps/e7g4eglj","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -837,7 +837,7 @@ f7c1c579-2dc0-47e2-ba19-8165c5a0e353(Pancreas),"step2:https://wandb.ai/xzy11632/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.25850000977516174,all_total_runs:600|all_finished_runs:457|step2_total_runs:540|step2_finished_runs:410|step3_0_total_runs:20|step3_0_finished_runs:16|step3_1_total_runs:20|step3_1_finished_runs:16|step3_2_total_runs:20|step3_2_finished_runs:15|finished_rate:76.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/alp04n2p|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/ki0b8xal,https://wandb.ai/xzy11632/dance-dev/sweeps/hzv5a32f,https://wandb.ai/xzy11632/dance-dev/sweeps/vvy7c4q9","- type: filter.gene
+",0.2585000097751617,all_total_runs:600|all_finished_runs:457|step2_total_runs:540|step2_finished_runs:410|step3_0_total_runs:20|step3_0_finished_runs:16|step3_1_total_runs:20|step3_1_finished_runs:16|step3_2_total_runs:20|step3_2_finished_runs:15|finished_rate:76.17%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/alp04n2p|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/ki0b8xal,https://wandb.ai/xzy11632/dance-dev/sweeps/hzv5a32f,https://wandb.ai/xzy11632/dance-dev/sweeps/vvy7c4q9","- type: filter.gene
   target: FilterGenesPercentile
   params: {}
 - type: normalize
@@ -862,7 +862,7 @@ f7c1c579-2dc0-47e2-ba19-8165c5a0e353(Pancreas),"step2:https://wandb.ai/xzy11632/
   params:
     config_dict:
       label_channel: cell_type
-",0.2599999904632568,all_total_runs:210|all_finished_runs:121|step2_total_runs:150|step2_finished_runs:104|step3_0_total_runs:20|step3_0_finished_runs:7|step3_1_total_runs:20|step3_1_finished_runs:6|step3_2_total_runs:20|step3_2_finished_runs:4|finished_rate:57.62%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/q7pjlbgz|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/hn65sj20,https://wandb.ai/xzy11632/dance-dev/sweeps/0aloe2j4,https://wandb.ai/xzy11632/dance-dev/sweeps/2fp77b7l","- type: filter.gene
+",0.25999999046325684,all_total_runs:210|all_finished_runs:121|step2_total_runs:150|step2_finished_runs:104|step3_0_total_runs:20|step3_0_finished_runs:7|step3_1_total_runs:20|step3_1_finished_runs:6|step3_2_total_runs:20|step3_2_finished_runs:4|finished_rate:57.62%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/q7pjlbgz|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/hn65sj20,https://wandb.ai/xzy11632/dance-dev/sweeps/0aloe2j4,https://wandb.ai/xzy11632/dance-dev/sweeps/2fp77b7l","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize
@@ -1255,7 +1255,7 @@ a68b64d8-aee3-4947-81b7-36b8fe5a44d2(Pancreas),"step2:https://wandb.ai/xzy11632/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.25850000977516174,all_total_runs:196|all_finished_runs:151|step2_total_runs:136|step2_finished_runs:91|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:77.04%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/ghnlphpt|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/qmwfsxt6,https://wandb.ai/xzy11632/dance-dev/sweeps/0bv97iz8,https://wandb.ai/xzy11632/dance-dev/sweeps/fhv7qkjx","- type: filter.gene
+",0.2585000097751617,all_total_runs:196|all_finished_runs:151|step2_total_runs:136|step2_finished_runs:91|step3_0_total_runs:20|step3_0_finished_runs:20|step3_1_total_runs:20|step3_1_finished_runs:20|step3_2_total_runs:20|step3_2_finished_runs:20|finished_rate:77.04%,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/ghnlphpt|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/qmwfsxt6,https://wandb.ai/xzy11632/dance-dev/sweeps/0bv97iz8,https://wandb.ai/xzy11632/dance-dev/sweeps/fhv7qkjx","- type: filter.gene
   target: FilterGenesScanpyOrder
   params:
     order:
@@ -1488,7 +1488,7 @@ fa27492b-82ff-4ab7-ac61-0e2b184eee67(Pancreas),"step2:https://wandb.ai/xzy11632/
     config_dict:
       feature_channel: feature.cell
       label_channel: cell_type
-",0.25850000977516174,all_total_runs:600|all_finished_runs:503|step2_total_runs:540|step2_finished_runs:448|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:83.83%,,,,,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/zcjk9kzu|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/xa1zrlvf,https://wandb.ai/xzy11632/dance-dev/sweeps/svep9r0f,https://wandb.ai/xzy11632/dance-dev/sweeps/earmglny","- type: filter.gene
+",0.2585000097751617,all_total_runs:600|all_finished_runs:503|step2_total_runs:540|step2_finished_runs:448|step3_0_total_runs:20|step3_0_finished_runs:18|step3_1_total_runs:20|step3_1_finished_runs:19|step3_2_total_runs:20|step3_2_finished_runs:18|finished_rate:83.83%,,,,,"step2:https://wandb.ai/xzy11632/dance-dev/sweeps/zcjk9kzu|step3:https://wandb.ai/xzy11632/dance-dev/sweeps/xa1zrlvf,https://wandb.ai/xzy11632/dance-dev/sweeps/svep9r0f,https://wandb.ai/xzy11632/dance-dev/sweeps/earmglny","- type: filter.gene
   target: FilterGenesPlaceHolder
   params: {}
 - type: normalize