get_group_id
Return an array of group identifiers for all objects.
Method call format
get_group_id()
Type of return value
list
Example
from catboost import Pool
train_data = [[75, 'today', 5, 4, 3],
[76, 'blvd', 4, 5, 7],
[70, 'letters', 0, 3, 4],
[60, 'back', 9, 0, 1]]
label_values = [0, 1, 0, 4]
input_pool = Pool(data=train_data,
label=label_values,
cat_features=[1],
group_id=["CA", "TN", "TN", "TN"])
print(input_pool.get_group_id())
Output:
[13555145920995255203L, 14627267897973961738L, 14627267897973961738L, 14627267897973961738L]