get_feature_names
Return the names of features from the dataset.
Method call format
get_feature_names()
Type of return value
List of strings.
Usage examples
Feature names are not set
import numpy as np
from catboost import FeaturesData
fd = FeaturesData(
num_feature_data=np.array([[1, 4, 5, 6], [4, 5, 6, 7], [30, 40, 50, 60]], dtype=np.float32),
cat_feature_data=np.array([["a", "b"], ["a", "b"], ["c", "d"]], dtype=object)
)
# print feature names
# the returned value is ['', '', '', '', '', ''] as neither num_feature_names nor cat_feature_names are specified
print(fd.get_feature_names())
Feature names are set
import numpy as np
from catboost import FeaturesData
fd = FeaturesData(
num_feature_data=np.array([[1, 4, 5, 6], [4, 5, 6, 7], [30, 40, 50, 60]], dtype=np.float32),
num_feature_names=['num_feat0', 'num_feat1', 'num_feat2', 'num_feat3'],
cat_feature_data=np.array([["a", "b"], ["a", "b"], ["c", "d"]], dtype=object),
cat_feature_names=['cat_feat0', 'cat_feat1']
)
# prints ['num_feat0', 'num_feat1', 'num_feat2', 'num_feat3', 'cat_feat0', 'cat_feat1']
print(fd.get_feature_names())