{ "data_id": "44458", "name": "compass_seed_0_nrows_2000_nclasses_10_ncols_100_stratify_True", "exact_name": "compass_seed_0_nrows_2000_nclasses_10_ncols_100_stratify_True", "version": 1, "version_label": "a4205c36-ca51-42f8-b040-397565870f04", "description": "Subsampling of the dataset compass (44162) with\n\nseed=0\nargs.nrows=2000\nargs.ncols=100\nargs.nclasses=10\nargs.no_stratify=True\nGenerated with the following source code:\n\n\n```python\n def subsample(\n self,\n seed: int,\n nrows_max: int = 2_000,\n ncols_max: int = 100,\n nclasses_max: int = 10,\n stratified: bool = True,\n ) -> Dataset:\n rng = np.random.default_rng(seed)\n\n x = self.x\n y = self.y\n\n # Uniformly sample\n classes = y.unique()\n if len(classes) > nclasses_max:\n vcs = y.value_counts()\n selected_classes = rng.choice(\n classes,\n size=nclasses_max,\n replace=False,\n p=vcs \/ sum(vcs),\n )\n\n # Select the indices where one of these classes is present\n idxs = y.index[y.isin(classes)]\n x = x.iloc[idxs]\n y = y.iloc[idxs]\n\n # Uniformly sample columns if required\n if len(x.columns) > ncols_max:\n columns_idxs = rng.choice(\n list(range(len(x.columns))), size=ncols_max, replace=False\n )\n sorted_column_idxs = sorted(columns_idxs)\n selected_columns = list(x.columns[sorted_column_idxs])\n x = x[selected_columns]\n else:\n sorted_column_idxs = list(range(len(x.columns)))\n\n if len(x) > nrows_max:\n # Stratify accordingly\n target_name = y.name\n data = pd.concat((x, y), axis=\"columns\")\n _, subset = train_test_split(\n data,\n test_size=nrows_max,\n stratify=data[target_name],\n shuffle=True,\n random_state=seed,\n )\n x = subset.drop(target_name, axis=\"columns\")\n y = subset[target_name]\n\n # We need to convert categorical columns to string for openml\n categorical_mask = [self.categorical_mask[i] for i in sorted_column_idxs]\n columns = list(x.columns)\n\n return Dataset(\n # Technically this is not the same but it's where it was derived from\n dataset=self.dataset,\n x=x,\n y=y,\n categorical_mask=categorical_mask,\n columns=columns,\n )\n```", "format": "arff", "uploader": "Eddie Bergman", "uploader_id": 32840, "visibility": "public", "creator": "\"Eddie Bergman\"", "contributor": null, "date": "2022-11-17 18:03:10", "update_comment": null, "last_update": "2022-11-17 18:03:10", "licence": "See source", "status": "active", "error_message": null, "url": "https:\/\/api.openml.org\/data\/download\/22111220\/dataset", "default_target_attribute": "is_recid", "row_id_attribute": null, "ignore_attribute": null, "runs": 0, "suggest": { "input": [ "compass_seed_0_nrows_2000_nclasses_10_ncols_100_stratify_True", "Subsampling of the dataset compass (44162) with seed=0 args.nrows=2000 args.ncols=100 args.nclasses=10 args.no_stratify=True Generated with the following source code: ```python def subsample( self, seed: int, nrows_max: int = 2_000, ncols_max: int = 100, nclasses_max: int = 10, stratified: bool = True, ) -> Dataset: rng = np.random.default_rng(seed) x = self.x y = self.y # Uniformly sample classes = y.unique() if len(classes) > nclasses_max: vcs = y.value_counts() selected_classes = rng.choice( " ], "weight": 5 }, "qualities": { "NumberOfInstances": 2000, "NumberOfFeatures": 18, "NumberOfClasses": 2, "NumberOfMissingValues": 0, "NumberOfInstancesWithMissingValues": 0, "NumberOfNumericFeatures": 8, "NumberOfSymbolicFeatures": 10, "PercentageOfBinaryFeatures": 11.11111111111111, "PercentageOfInstancesWithMissingValues": 0, "AutoCorrelation": 0.5002501250625313, "PercentageOfMissingValues": 0, "Dimensionality": 0.009, "PercentageOfNumericFeatures": 44.44444444444444, "MajorityClassPercentage": 50, "PercentageOfSymbolicFeatures": 55.55555555555556, "MajorityClassSize": 1000, "MinorityClassPercentage": 50, "MinorityClassSize": 1000, "NumberOfBinaryFeatures": 2 }, "tags": [ { "uploader": "38960", "tag": "Machine Learning" }, { "uploader": "38960", "tag": "Mathematics" } ], "features": [ { "name": "is_recid", "index": "17", "type": "nominal", "distinct": "2", "missing": "0", "target": "1", "distr": [ [ "-1", "0" ], [ [ "1000", "0" ], [ "0", "1000" ] ] ] }, { "name": "sex", "index": "0", "type": "nominal", "distinct": "2", "missing": "0", "distr": [ [ "0", "1" ], [ [ "221", "148" ], [ "779", "852" ] ] ] }, { "name": "age", "index": "1", "type": "numeric", "distinct": "57", "missing": "0", "min": "18", "max": "78", "mean": "34", "stdev": "11" }, { "name": "age_cat", "index": "2", "type": "nominal", "distinct": "3", "missing": "0", "distr": [ [ "0", "1", "2" ], [ [ "598", "568" ], [ "220", "154" ], [ "182", "278" ] ] ] }, { "name": "race", "index": "3", "type": "nominal", "distinct": "6", "missing": "0", "distr": [ [ "0", "1", "2", "3", "4", "5" ], [ [ "448", "602" ], [ "9", "2" ], [ "387", "302" ], [ "94", "54" ], [ "3", "2" ], [ "59", "38" ] ] ] }, { "name": "juv_fel_count", "index": "4", "type": "numeric", "distinct": "6", "missing": "0", "min": "0", "max": "5", "mean": "0", "stdev": "0" }, { "name": "juv_misd_count", "index": "5", "type": "numeric", "distinct": "9", "missing": "0", "min": "0", "max": "12", "mean": "0", "stdev": "1" }, { "name": "juv_other_count", "index": "6", "type": "numeric", "distinct": "6", "missing": "0", "min": "0", "max": "5", "mean": "0", "stdev": "0" }, { "name": "priors_count", "index": "7", "type": "numeric", "distinct": "34", "missing": "0", "min": "0", "max": "36", "mean": "4", "stdev": "6" }, { "name": "days_b_screening_arrest", "index": "8", "type": "numeric", "distinct": "205", "missing": "0", "min": "-414", "max": "1026", "mean": "5", "stdev": "79" }, { "name": "c_days_from_compas", "index": "9", "type": "numeric", "distinct": "189", "missing": "0", "min": "0", "max": "9485", "mean": "44", "stdev": "322" }, { "name": "c_charge_degree", "index": "10", "type": "nominal", "distinct": "11", "missing": "0", "distr": [ [ "0", "1", "10", "11", "12", "2", "3", "4", "5", "6", "7", "8", "9" ], [ [ "0", "1" ], [ "25", "9" ], [ "0", "2" ], [ "0", "0" ], [ "0", "0" ], [ "100", "81" ], [ "568", "627" ], [ "1", "0" ], [ "3", "0" ], [ "5", "6" ], [ "235", "209" ], [ "60", "59" ], [ "3", "6" ] ] ] }, { "name": "decile_score.1", "index": "11", "type": "nominal", "distinct": "10", "missing": "0", "distr": [ [ "0", "1", "2", "3", "4", "5", "6", "7", "8", "9" ], [ [ "251", "62" ], [ "154", "75" ], [ "114", "86" ], [ "103", "100" ], [ "73", "91" ], [ "72", "126" ], [ "66", "116" ], [ "61", "129" ], [ "59", "115" ], [ "47", "100" ] ] ] }, { "name": "score_text", "index": "12", "type": "nominal", "distinct": "3", "missing": "0", "distr": [ [ "0", "1", "2" ], [ [ "167", "344" ], [ "622", "323" ], [ "211", "333" ] ] ] }, { "name": "v_type_of_assessment", "index": "13", "type": "nominal", "distinct": "1", "missing": "0", "distr": [ [ "0" ], [ [ "1000", "1000" ] ] ] }, { "name": "v_decile_score", "index": "14", "type": "nominal", "distinct": "10", "missing": "0", "distr": [ [ "0", "1", "2", "3", "4", "5", "6", "7", "8", "9" ], [ [ "321", "114" ], [ "185", "116" ], [ "115", "134" ], [ "95", "138" ], [ "72", "112" ], [ "79", "112" ], [ "49", "108" ], [ "39", "65" ], [ "27", "64" ], [ "18", "37" ] ] ] }, { "name": "v_score_text", "index": "15", "type": "nominal", "distinct": "3", "missing": "0", "distr": [ [ "0", "1", "2" ], [ [ "84", "166" ], [ "716", "502" ], [ "200", "332" ] ] ] }, { "name": "end", "index": "16", "type": "numeric", "distinct": "931", "missing": "0", "min": "0", "max": "1186", "mean": "615", "stdev": "339" } ], "nr_of_issues": 0, "nr_of_downvotes": 0, "nr_of_likes": 0, "nr_of_downloads": 0, "total_downloads": 0, "reach": 0, "reuse": 0, "impact_of_reuse": 0, "reach_of_reuse": 0, "impact": 0 }