np.average векторов слов - PullRequest
       0

np.average векторов слов

0 голосов
/ 15 февраля 2020

У меня есть этот словарь, который содержит слова в качестве ключей и их векторы в качестве значений.

my_dict = {'new': array([ 6.77980e-02, -2.07800e-02, -1.22845e-01, 1.75853e-01,  1.49210e-02]), 
           'its': array([ 7.85300e-03, -8.81160e-02,  2.60125e-01,  1.77740e-02, -1.09075e-011])}

Я хотел бы рассчитать поэлементное среднее векторов для этих данных слов, используя np.average(). Результат должен быть np.ndarray.

Это моя попытка:

average = [np.average(my_dict[x], axis=None) for x in self.my_dict]

1 Ответ

0 голосов
/ 15 февраля 2020

Чтобы вычислить одно среднее значение по всем значениям (а не собрать набор бессмысленных средних для каждого отдельного вектора), попробуйте:

np.average(list(my_dict.values()))

(если он не возвращает ожидаемую форму , попробуйте каждую явную axis= возможность.)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...