Мы можем добавить функции графа в нашу модель машинного обучения, такие как центральность собственного вектора. Например, предположим, что мы хотим предсказать, пройдет ли студент курс на основе своего учителя и консультанта. В приведенном ниже классе Python мы рассчитываем центральность среди учителей и консультантов. Аналогичным образом мы заменяем эти категориальные уровни значениями центральности, хотя мы могли бы легко изменить код, добавив столбец и оставив исходные значения нетронутыми. Следует отметить, что эта реализация имеет небольшую проблему утечки, поскольку она включает в себя каждую строку в вычислении центральности. Опять же, это можно настроить по желанию. Наконец, приведенный ниже класс можно включить в конвейер scikit-learn.