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]