feat: MP-20 integration

.gitignore

@@ -1,5 +1,6 @@
 hydra-outputs/
 multirun/
+*.zip
 
 .vscode/
 # Byte-compiled / optimized / DLL files
@@ -217,4 +218,7 @@ results_final_before_and_after_relax.zip
 command.txt
 results_october/
 results_before_and_after_relax/
-results_neurips/
+results_neurips/
+
+baseline_data/
+baseline_results/

data/mp20_composition_counts_distribution.json

@@ -0,0 +1,120 @@
+{
+    "1": 0.010079966667200846,
+    "2": 1.9230461050303682e-05,
+    "3": 0.024316917998109006,
+    "4": 0.002310860402878159,
+    "5": 0.012323520456402942,
+    "6": 0.007605647345395106,
+    "7": 0.009589589910418103,
+    "8": 0.1866668803384561,
+    "9": 0.04778128555632121,
+    "10": 3.20507684171728e-06,
+    "11": 0.00882678162208939,
+    "12": 0.013249787663659236,
+    "13": 0.01771125462732969,
+    "14": 0.015871540520183972,
+    "15": 0.010243425586128427,
+    "16": 0.03568853063252191,
+    "17": 0.02077530808801141,
+    "18": 6.41015368343456e-06,
+    "19": 0.010166503741927213,
+    "20": 0.009326773609397286,
+    "21": 0.005499911860386853,
+    "22": 0.009983814361949327,
+    "23": 0.011660069550167466,
+    "24": 0.007913334722199964,
+    "25": 0.0176759987820708,
+    "26": 0.021227223922693546,
+    "27": 0.01970481242287784,
+    "28": 0.022028493133122867,
+    "29": 0.020531722248040896,
+    "30": 0.01084277495552956,
+    "31": 0.0154132145318184,
+    "32": 0.01707664941266967,
+    "33": 0.00787166872325764,
+    "34": 0.021900290059454177,
+    "35": 0.011929296004871717,
+    "36": 2.243553789202096e-05,
+    "37": 0.008458197785291902,
+    "38": 0.01126584509863624,
+    "39": 0.008294738866364322,
+    "40": 0.007147321357029535,
+    "41": 0.0073748818127914615,
+    "42": 0.007221038124389032,
+    "43": 0.0013685678114132786,
+    "44": 0.006262720148715565,
+    "45": 0.009432541145173955,
+    "46": 0.010845980032371276,
+    "47": 0.009137674075735966,
+    "48": 0.005910161696126665,
+    "49": 0.011474175093347864,
+    "50": 0.01205429400169869,
+    "51": 0.010788288649220365,
+    "52": 0.01461194532138908,
+    "53": 0.011224179099693915,
+    "54": 4.487107578404192e-05,
+    "55": 0.008660117626320091,
+    "56": 0.011496610631239884,
+    "57": 0.01009919712825115,
+    "58": 0.007227448278072467,
+    "59": 0.006519126296052948,
+    "60": 0.006919760901267608,
+    "61": 0.002173042098684316,
+    "62": 0.006394128299225974,
+    "63": 0.00517299402253169,
+    "64": 0.003762760212176087,
+    "65": 0.006147337382413743,
+    "66": 0.006214643996089806,
+    "67": 0.006179388150830916,
+    "68": 0.006503100911844362,
+    "69": 0.005035175718337847,
+    "70": 0.004538388807871669,
+    "71": 0.0048781269530937,
+    "72": 0.004483902501562475,
+    "73": 0.004535183731029952,
+    "74": 0.00463133603628147,
+    "75": 0.002128171022900274,
+    "76": 0.002528805628114934,
+    "77": 0.005964648002435858,
+    "78": 0.008102434255861284,
+    "79": 0.00780115703273986,
+    "80": 0.00517299402253169,
+    "81": 0.006922965978109325,
+    "82": 0.005330042787775837,
+    "83": 0.007265909200173074,
+    "84": 0.0,
+    "85": 0.0,
+    "86": 0.0,
+    "87": 0.0,
+    "88": 0.0,
+    "89": 0.0011249819714427654,
+    "90": 0.0028204676207112065,
+    "91": 0.0007339625967532572,
+    "92": 0.004490312655245909,
+    "93": 0.0005096072178330475,
+    "94": 0.0007692184420121473,
+    "95": 0.0,
+    "96": 0.0,
+    "97": 0.0,
+    "98": 0.0,
+    "99": 0.0,
+    "100": 0.0,
+    "101": 0.0,
+    "102": 0.0,
+    "103": 0.0,
+    "104": 0.0,
+    "105": 0.0,
+    "106": 0.0,
+    "107": 0.0,
+    "108": 0.0,
+    "109": 0.0,
+    "110": 0.0,
+    "111": 0.0,
+    "112": 0.0,
+    "113": 0.0,
+    "114": 0.0,
+    "115": 0.0,
+    "116": 0.0,
+    "117": 0.0,
+    "118": 0.0
+}

data/mp20_composition_distribution.json

@@ -0,0 +1,120 @@
+{
+    "1": 0.008220062824765876,
+    "2": 4.403605084696004e-05,
+    "3": 0.02870416581041012,
+    "4": 0.0022384992513871356,
+    "5": 0.009959486833220796,
+    "6": 0.009438393564865102,
+    "7": 0.010098934327569503,
+    "8": 0.08456389630977894,
+    "9": 0.02332442826527317,
+    "10": 7.339341807826674e-06,
+    "11": 0.013144761177817573,
+    "12": 0.018135513607139712,
+    "13": 0.01803276282183014,
+    "14": 0.018465783988491913,
+    "15": 0.011823679652408771,
+    "16": 0.02367671667204885,
+    "17": 0.01222734345183924,
+    "18": 1.4678683615653349e-05,
+    "19": 0.013959428118486335,
+    "20": 0.013886034700408067,
+    "21": 0.008095294014032821,
+    "22": 0.0129319202653906,
+    "23": 0.013386959457475854,
+    "24": 0.010311775239996476,
+    "25": 0.019522649208818955,
+    "26": 0.018920823180577164,
+    "27": 0.019309808296391978,
+    "28": 0.02228958107036961,
+    "29": 0.024931744121187212,
+    "30": 0.013878695358600241,
+    "31": 0.015625458708862987,
+    "32": 0.01756304494612923,
+    "33": 0.009680591844523382,
+    "34": 0.016763056689076124,
+    "35": 0.0077576842908727945,
+    "36": 5.137539265478672e-05,
+    "37": 0.011177817573320025,
+    "38": 0.015537386607169069,
+    "39": 0.01211725332472184,
+    "40": 0.008829228194815488,
+    "41": 0.008579690573349383,
+    "42": 0.008799870827584183,
+    "43": 0.0020623550479992956,
+    "44": 0.008557672547925901,
+    "45": 0.010854886533775652,
+    "46": 0.013298887355781934,
+    "47": 0.0137025511552124,
+    "48": 0.009981504858644278,
+    "49": 0.01525115227666383,
+    "50": 0.014847488477233361,
+    "51": 0.01313008249420192,
+    "52": 0.0126016498840384,
+    "53": 0.007317323782403194,
+    "54": 9.541144350174676e-05,
+    "55": 0.01165487479082876,
+    "56": 0.015772245545019523,
+    "57": 0.013974106802101987,
+    "58": 0.01081085048292869,
+    "59": 0.00933564277955553,
+    "60": 0.010135631036608637,
+    "61": 0.0033247218389454835,
+    "62": 0.009614537768252943,
+    "63": 0.008381528344538062,
+    "64": 0.006098993042303966,
+    "65": 0.008139330064879782,
+    "66": 0.008733816751313742,
+    "67": 0.008425564395385022,
+    "68": 0.008777852802160702,
+    "69": 0.007287966415171887,
+    "70": 0.007302645098787541,
+    "71": 0.006884302615741421,
+    "72": 0.006106332384111793,
+    "73": 0.006451281449079646,
+    "74": 0.0077576842908727945,
+    "75": 0.003170595660981123,
+    "76": 0.003823797081877697,
+    "77": 0.007882453101605848,
+    "78": 0.009900772098758183,
+    "79": 0.010333793265419958,
+    "80": 0.007309984440595368,
+    "81": 0.010201685112879076,
+    "82": 0.007706308898218008,
+    "83": 0.010304435898188651,
+    "84": 0.0,
+    "85": 0.0,
+    "86": 0.0,
+    "87": 0.0,
+    "88": 0.0,
+    "89": 0.0017247453248392685,
+    "90": 0.004212782197692511,
+    "91": 0.0014972257287966416,
+    "92": 0.006832927223086633,
+    "93": 0.0010275078530957344,
+    "94": 0.001372456918063588,
+    "95": 0.0,
+    "96": 0.0,
+    "97": 0.0,
+    "98": 0.0,
+    "99": 0.0,
+    "100": 0.0,
+    "101": 0.0,
+    "102": 0.0,
+    "103": 0.0,
+    "104": 0.0,
+    "105": 0.0,
+    "106": 0.0,
+    "107": 0.0,
+    "108": 0.0,
+    "109": 0.0,
+    "110": 0.0,
+    "111": 0.0,
+    "112": 0.0,
+    "113": 0.0,
+    "114": 0.0,
+    "115": 0.0,
+    "116": 0.0,
+    "117": 0.0,
+    "118": 0.0
+}