MCPcopy Create free account
hub / github.com/antmachineintelligence/mtgbmcode / subset

Method subset

python-package/lightgbmmt/basic.py:1130–1153  ·  view source on GitHub ↗

Get subset of current Dataset. Parameters ---------- used_indices : list of int Indices used to create the subset. params : dict or None, optional (default=None) These parameters will be passed to Dataset constructor. Returns

(self, used_indices, params=None)

Source from the content-addressed store, hash-verified

1128 return ret
1129
1130 def subset(self, used_indices, params=None):
1131 """Get subset of current Dataset.
1132
1133 Parameters
1134 ----------
1135 used_indices : list of int
1136 Indices used to create the subset.
1137 params : dict or None, optional (default=None)
1138 These parameters will be passed to Dataset constructor.
1139
1140 Returns
1141 -------
1142 subset : Dataset
1143 Subset of the current Dataset.
1144 """
1145 if params is None:
1146 params = self.params
1147 ret = Dataset(None, reference=self, feature_name=self.feature_name,
1148 categorical_feature=self.categorical_feature, params=params,
1149 free_raw_data=self.free_raw_data)
1150 ret._predictor = self._predictor
1151 ret.pandas_categorical = self.pandas_categorical
1152 ret.used_indices = sorted(used_indices)
1153 return ret
1154
1155 def save_binary(self, filename):
1156 """Save Dataset to a binary file.

Callers 4

test_subset_groupMethod · 0.95
test_reference_chainMethod · 0.95
test_init_with_subsetMethod · 0.95
_make_n_foldsFunction · 0.80

Calls 1

DatasetClass · 0.70

Tested by 3

test_subset_groupMethod · 0.76
test_reference_chainMethod · 0.76
test_init_with_subsetMethod · 0.76