Проще говоря, train_index выполняет алгоритм кластеризации векторов вашей базы данных и сохраняет центроиды кластера в вашем индексе.