--- a
+++ b/datasets/SUSTech1K/SUSTech1K.json
@@ -0,0 +1,1056 @@
+{
+    "TRAIN_SET": [
+        "0000",
+        "0002",
+        "0005",
+        "0018",
+        "0021",
+        "0026",
+        "0033",
+        "0039",
+        "0040",
+        "0044",
+        "0046",
+        "0047",
+        "0050",
+        "0052",
+        "0060",
+        "0062",
+        "0066",
+        "0073",
+        "0075",
+        "0078",
+        "0080",
+        "0092",
+        "0093",
+        "0096",
+        "0097",
+        "0102",
+        "0103",
+        "0116",
+        "0118",
+        "0126",
+        "0128",
+        "0144",
+        "0149",
+        "0151",
+        "0153",
+        "0154",
+        "0156",
+        "0157",
+        "0158",
+        "0164",
+        "0165",
+        "0168",
+        "0169",
+        "0174",
+        "0180",
+        "0183",
+        "0193",
+        "0194",
+        "0196",
+        "0199",
+        "0201",
+        "0203",
+        "0204",
+        "0208",
+        "0212",
+        "0220",
+        "0226",
+        "0227",
+        "0231",
+        "0232",
+        "0243",
+        "0253",
+        "0257",
+        "0267",
+        "0268",
+        "0282",
+        "0284",
+        "0287",
+        "0289",
+        "0291",
+        "0293",
+        "0295",
+        "0297",
+        "0301",
+        "0306",
+        "0311",
+        "0315",
+        "0318",
+        "0320",
+        "0321",
+        "0323",
+        "0325",
+        "0327",
+        "0341",
+        "0347",
+        "0351",
+        "0356",
+        "0357",
+        "0360",
+        "0367",
+        "0372",
+        "0379",
+        "0380",
+        "0395",
+        "0402",
+        "0412",
+        "0421",
+        "0425",
+        "0434",
+        "0438",
+        "0440",
+        "0441",
+        "0444",
+        "0452",
+        "0457",
+        "0458",
+        "0464",
+        "0468",
+        "0473",
+        "0474",
+        "0478",
+        "0488",
+        "0493",
+        "0495",
+        "0496",
+        "0497",
+        "0502",
+        "0514",
+        "0522",
+        "0521",
+        "0525",
+        "0533",
+        "0535",
+        "0537",
+        "0540",
+        "0544",
+        "0545",
+        "0546",
+        "0547",
+        "0551",
+        "0552",
+        "0553",
+        "0555",
+        "0557",
+        "0559",
+        "0577",
+        "0581",
+        "0583",
+        "0584",
+        "0585",
+        "0591",
+        "0597",
+        "0600",
+        "0605",
+        "0610",
+        "0611",
+        "0612",
+        "0616",
+        "0631",
+        "0632",
+        "0634",
+        "0636",
+        "0637",
+        "0641",
+        "0649",
+        "0653",
+        "0655",
+        "0664",
+        "0665",
+        "0671",
+        "0675",
+        "0677",
+        "0687",
+        "0695",
+        "0701",
+        "0702",
+        "0707",
+        "0717",
+        "0720",
+        "0721",
+        "0723",
+        "0726",
+        "0731",
+        "0756",
+        "0757",
+        "0759",
+        "0760",
+        "0767",
+        "0770",
+        "0773",
+        "0779",
+        "0780",
+        "0783",
+        "0791",
+        "0792",
+        "0796",
+        "0805",
+        "0810",
+        "0811",
+        "0823",
+        "0828",
+        "0830",
+        "0839",
+        "0841",
+        "0844",
+        "0845",
+        "0846",
+        "0850",
+        "0853",
+        "0860",
+        "0862",
+        "0863",
+        "0865",
+        "0868",
+        "0869",
+        "0876",
+        "0883",
+        "0884",
+        "0888",
+        "0897",
+        "0904",
+        "0906",
+        "0907",
+        "0908",
+        "0918",
+        "0922",
+        "0923",
+        "0925",
+        "0933",
+        "0938",
+        "0942",
+        "0943",
+        "0944",
+        "0948",
+        "0951",
+        "0959",
+        "0965",
+        "0966",
+        "0969",
+        "0970",
+        "0973",
+        "0978",
+        "0979",
+        "0982",
+        "0996",
+        "0997",
+        "1002",
+        "1004",
+        "1011",
+        "1013",
+        "1015",
+        "1019",
+        "1024",
+        "1026",
+        "1027",
+        "1036",
+        "1038",
+        "1046",
+        "1056",
+        "1057"
+    ],
+    "TEST_SET": [
+        "0001",
+        "0003",
+        "0004",
+        "0006",
+        "0007",
+        "0008",
+        "0009",
+        "0010",
+        "0011",
+        "0012",
+        "0013",
+        "0014",
+        "0015",
+        "0016",
+        "0017",
+        "0019",
+        "0020",
+        "0022",
+        "0023",
+        "0024",
+        "0025",
+        "0027",
+        "0028",
+        "0029",
+        "0030",
+        "0031",
+        "0032",
+        "0034",
+        "0035",
+        "0036",
+        "0037",
+        "0038",
+        "0041",
+        "0042",
+        "0043",
+        "0045",
+        "0048",
+        "0049",
+        "0051",
+        "0053",
+        "0054",
+        "0055",
+        "0056",
+        "0057",
+        "0058",
+        "0059",
+        "0061",
+        "0063",
+        "0064",
+        "0065",
+        "0067",
+        "0068",
+        "0069",
+        "0070",
+        "0071",
+        "0072",
+        "0074",
+        "0076",
+        "0077",
+        "0079",
+        "0081",
+        "0082",
+        "0083",
+        "0084",
+        "0085",
+        "0086",
+        "0087",
+        "0088",
+        "0089",
+        "0090",
+        "0091",
+        "0094",
+        "0095",
+        "0098",
+        "0099",
+        "0100",
+        "0101",
+        "0104",
+        "0105",
+        "0106",
+        "0107",
+        "0108",
+        "0109",
+        "0110",
+        "0111",
+        "0112",
+        "0113",
+        "0114",
+        "0115",
+        "0117",
+        "0119",
+        "0120",
+        "0121",
+        "0122",
+        "0123",
+        "0124",
+        "0125",
+        "0127",
+        "0129",
+        "0130",
+        "0131",
+        "0132",
+        "0133",
+        "0134",
+        "0135",
+        "0136",
+        "0137",
+        "0138",
+        "0139",
+        "0140",
+        "0141",
+        "0142",
+        "0143",
+        "0145",
+        "0146",
+        "0147",
+        "0148",
+        "0150",
+        "0152",
+        "0155",
+        "0159",
+        "0160",
+        "0161",
+        "0162",
+        "0163",
+        "0166",
+        "0167",
+        "0170",
+        "0171",
+        "0172",
+        "0173",
+        "0175",
+        "0176",
+        "0177",
+        "0178",
+        "0179",
+        "0181",
+        "0182",
+        "0184",
+        "0185",
+        "0186",
+        "0187",
+        "0188",
+        "0189",
+        "0190",
+        "0191",
+        "0192",
+        "0195",
+        "0197",
+        "0198",
+        "0200",
+        "0202",
+        "0205",
+        "0206",
+        "0207",
+        "0209",
+        "0210",
+        "0211",
+        "0213",
+        "0214",
+        "0215",
+        "0216",
+        "0217",
+        "0218",
+        "0219",
+        "0221",
+        "0222",
+        "0223",
+        "0224",
+        "0225",
+        "0228",
+        "0229",
+        "0230",
+        "0233",
+        "0234",
+        "0235",
+        "0236",
+        "0237",
+        "0238",
+        "0239",
+        "0240",
+        "0241",
+        "0242",
+        "0244",
+        "0245",
+        "0246",
+        "0247",
+        "0248",
+        "0249",
+        "0250",
+        "0251",
+        "0252",
+        "0254",
+        "0255",
+        "0256",
+        "0258",
+        "0259",
+        "0260",
+        "0261",
+        "0262",
+        "0263",
+        "0264",
+        "0265",
+        "0266",
+        "0269",
+        "0270",
+        "0271",
+        "0272",
+        "0273",
+        "0274",
+        "0275",
+        "0276",
+        "0277",
+        "0278",
+        "0279",
+        "0280",
+        "0281",
+        "0283",
+        "0285",
+        "0286",
+        "0288",
+        "0290",
+        "0292",
+        "0294",
+        "0296",
+        "0298",
+        "0299",
+        "0300",
+        "0302",
+        "0303",
+        "0304",
+        "0305",
+        "0307",
+        "0308",
+        "0309",
+        "0310",
+        "0312",
+        "0313",
+        "0314",
+        "0316",
+        "0317",
+        "0319",
+        "0322",
+        "0324",
+        "0326",
+        "0328",
+        "0329",
+        "0330",
+        "0331",
+        "0332",
+        "0333",
+        "0334",
+        "0335",
+        "0336",
+        "0337",
+        "0338",
+        "0339",
+        "0340",
+        "0342",
+        "0343",
+        "0344",
+        "0345",
+        "0346",
+        "0348",
+        "0349",
+        "0350",
+        "0353",
+        "0354",
+        "0355",
+        "0358",
+        "0359",
+        "0361",
+        "0362",
+        "0363",
+        "0364",
+        "0365",
+        "0366",
+        "0368",
+        "0369",
+        "0370",
+        "0371",
+        "0373",
+        "0374",
+        "0375",
+        "0376",
+        "0377",
+        "0378",
+        "0381",
+        "0382",
+        "0383",
+        "0384",
+        "0385",
+        "0386",
+        "0387",
+        "0388",
+        "0389",
+        "0390",
+        "0391",
+        "0392",
+        "0393",
+        "0394",
+        "0396",
+        "0397",
+        "0398",
+        "0399",
+        "0400",
+        "0401",
+        "0403",
+        "0404",
+        "0405",
+        "0406",
+        "0407",
+        "0408",
+        "0409",
+        "0410",
+        "0411",
+        "0413",
+        "0414",
+        "0415",
+        "0416",
+        "0417",
+        "0418",
+        "0419",
+        "0420",
+        "0422",
+        "0423",
+        "0424",
+        "0426",
+        "0427",
+        "0428",
+        "0429",
+        "0430",
+        "0431",
+        "0432",
+        "0433",
+        "0435",
+        "0436",
+        "0437",
+        "0439",
+        "0442",
+        "0443",
+        "0445",
+        "0446",
+        "0447",
+        "0448",
+        "0449",
+        "0450",
+        "0451",
+        "0453",
+        "0454",
+        "0455",
+        "0456",
+        "0459",
+        "0460",
+        "0461",
+        "0462",
+        "0463",
+        "0465",
+        "0466",
+        "0467",
+        "0469",
+        "0470",
+        "0471",
+        "0472",
+        "0475",
+        "0476",
+        "0477",
+        "0479",
+        "0480",
+        "0481",
+        "0482",
+        "0483",
+        "0484",
+        "0485",
+        "0486",
+        "0487",
+        "0489",
+        "0490",
+        "0491",
+        "0492",
+        "0494",
+        "0498",
+        "0499",
+        "0500",
+        "0501",
+        "0503",
+        "0504",
+        "0505",
+        "0506",
+        "0507",
+        "0508",
+        "0509",
+        "0510",
+        "0511",
+        "0512",
+        "0513",
+        "0515",
+        "0516",
+        "0517",
+        "0518",
+        "0519",
+        "0520",
+        "0523",
+        "0524",
+        "0526",
+        "0527",
+        "0528",
+        "0529",
+        "0530",
+        "0531",
+        "0532",
+        "0534",
+        "0536",
+        "0538",
+        "0539",
+        "0541",
+        "0542",
+        "0543",
+        "0548",
+        "0549",
+        "0550",
+        "0554",
+        "0556",
+        "0558",
+        "0560",
+        "0561",
+        "0562",
+        "0563",
+        "0564",
+        "0565",
+        "0566",
+        "0567",
+        "0568",
+        "0569",
+        "0570",
+        "0571",
+        "0572",
+        "0573",
+        "0574",
+        "0575",
+        "0576",
+        "0578",
+        "0579",
+        "0580",
+        "0582",
+        "0586",
+        "0587",
+        "0588",
+        "0589",
+        "0590",
+        "0592",
+        "0593",
+        "0594",
+        "0595",
+        "0596",
+        "0598",
+        "0599",
+        "0601",
+        "0602",
+        "0603",
+        "0604",
+        "0606",
+        "0607",
+        "0608",
+        "0609",
+        "0613",
+        "0614",
+        "0615",
+        "0617",
+        "0618",
+        "0619",
+        "0620",
+        "0621",
+        "0622",
+        "0623",
+        "0624",
+        "0625",
+        "0626",
+        "0627",
+        "0628",
+        "0629",
+        "0630",
+        "0633",
+        "0635",
+        "0638",
+        "0639",
+        "0640",
+        "0642",
+        "0643",
+        "0644",
+        "0645",
+        "0646",
+        "0647",
+        "0648",
+        "0650",
+        "0651",
+        "0652",
+        "0654",
+        "0656",
+        "0657",
+        "0658",
+        "0659",
+        "0660",
+        "0661",
+        "0662",
+        "0663",
+        "0666",
+        "0667",
+        "0668",
+        "0669",
+        "0670",
+        "0672",
+        "0673",
+        "0674",
+        "0676",
+        "0678",
+        "0679",
+        "0680",
+        "0681",
+        "0682",
+        "0683",
+        "0684",
+        "0685",
+        "0686",
+        "0688",
+        "0689",
+        "0690",
+        "0691",
+        "0692",
+        "0693",
+        "0694",
+        "0696",
+        "0697",
+        "0698",
+        "0699",
+        "0700",
+        "0703",
+        "0704",
+        "0705",
+        "0706",
+        "0708",
+        "0709",
+        "0710",
+        "0711",
+        "0712",
+        "0713",
+        "0714",
+        "0715",
+        "0716",
+        "0718",
+        "0719",
+        "0722",
+        "0724",
+        "0725",
+        "0727",
+        "0728",
+        "0729",
+        "0730",
+        "0732",
+        "0733",
+        "0734",
+        "0735",
+        "0736",
+        "0737",
+        "0738",
+        "0739",
+        "0740",
+        "0741",
+        "0742",
+        "0743",
+        "0744",
+        "0745",
+        "0746",
+        "0747",
+        "0748",
+        "0749",
+        "0750",
+        "0751",
+        "0752",
+        "0753",
+        "0754",
+        "0755",
+        "0758",
+        "0761",
+        "0762",
+        "0763",
+        "0764",
+        "0765",
+        "0766",
+        "0768",
+        "0769",
+        "0771",
+        "0772",
+        "0774",
+        "0775",
+        "0776",
+        "0777",
+        "0778",
+        "0781",
+        "0782",
+        "0784",
+        "0785",
+        "0786",
+        "0787",
+        "0788",
+        "0789",
+        "0790",
+        "0793",
+        "0794",
+        "0795",
+        "0797",
+        "0798",
+        "0799",
+        "0800",
+        "0801",
+        "0802",
+        "0803",
+        "0804",
+        "0806",
+        "0807",
+        "0808",
+        "0809",
+        "0812",
+        "0813",
+        "0814",
+        "0815",
+        "0816",
+        "0817",
+        "0818",
+        "0819",
+        "0820",
+        "0821",
+        "0822",
+        "0824",
+        "0825",
+        "0826",
+        "0827",
+        "0829",
+        "0831",
+        "0832",
+        "0833",
+        "0834",
+        "0835",
+        "0836",
+        "0837",
+        "0838",
+        "0840",
+        "0842",
+        "0843",
+        "0847",
+        "0848",
+        "0849",
+        "0851",
+        "0852",
+        "0854",
+        "0855",
+        "0856",
+        "0857",
+        "0858",
+        "0859",
+        "0861",
+        "0864",
+        "0866",
+        "0867",
+        "0870",
+        "0871",
+        "0872",
+        "0873",
+        "0874",
+        "0875",
+        "0877",
+        "0878",
+        "0879",
+        "0880",
+        "0881",
+        "0882",
+        "0885",
+        "0886",
+        "0887",
+        "0889",
+        "0890",
+        "0891",
+        "0892",
+        "0893",
+        "0894",
+        "0895",
+        "0896",
+        "0898",
+        "0899",
+        "0900",
+        "0901",
+        "0902",
+        "0903",
+        "0905",
+        "0909",
+        "0910",
+        "0911",
+        "0912",
+        "0913",
+        "0914",
+        "0915",
+        "0916",
+        "0917",
+        "0919",
+        "0920",
+        "0921",
+        "0924",
+        "0926",
+        "0927",
+        "0928",
+        "0929",
+        "0930",
+        "0931",
+        "0932",
+        "0934",
+        "0935",
+        "0936",
+        "0937",
+        "0939",
+        "0940",
+        "0941",
+        "0945",
+        "0946",
+        "0947",
+        "0949",
+        "0950",
+        "0952",
+        "0953",
+        "0954",
+        "0955",
+        "0956",
+        "0957",
+        "0958",
+        "0960",
+        "0961",
+        "0962",
+        "0963",
+        "0964",
+        "0967",
+        "0968",
+        "0971",
+        "0972",
+        "0974",
+        "0975",
+        "0976",
+        "0977",
+        "0980",
+        "0981",
+        "0983",
+        "0984",
+        "0985",
+        "0986",
+        "0987",
+        "0988",
+        "0989",
+        "0990",
+        "0991",
+        "0992",
+        "0993",
+        "0994",
+        "0995",
+        "0998",
+        "0999",
+        "1000",
+        "1001",
+        "1003",
+        "1005",
+        "1006",
+        "1007",
+        "1008",
+        "1009",
+        "1010",
+        "1012",
+        "1014",
+        "1016",
+        "1017",
+        "1018",
+        "1020",
+        "1021",
+        "1022",
+        "1023",
+        "1025",
+        "1028",
+        "1029",
+        "1030",
+        "1031",
+        "1032",
+        "1033",
+        "1034",
+        "1035",
+        "1037",
+        "1039",
+        "1040",
+        "1041",
+        "1042",
+        "1043",
+        "1044",
+        "1045",
+        "1049",
+        "1055"
+    ]
+}
\ No newline at end of file