# Choose the feature of interest (e.g., feature at index 0)
feature_index = 0