【发布时间】:2020-10-27 00:30:43
【问题描述】:
def cast_vector(row):
return np.array(list(map(lambda x: x.astype('float32'), row)))
words = pd.DataFrame(word_vectors.vocab.keys())
words.columns = ['words']
words['vectors'] = words.words.apply(lambda x: word_vectors.wv[f'{x}'])
words['vectors_typed'] = words.vectors.apply(cast_vector)
words['cluster'] = words.vectors_typed.apply(lambda x: model.predict([np.array(x)]))
#words.cluster = words.cluster.apply(lambda x: x[0])
为什么是float32却报错?
【问题讨论】:
-
看起来您的模型需要一个 float64 (double) 值;如果您根本不转换为 float32 那么它应该可以工作。
-
我现在尝试将其设置为双倍值,但仍然无法正常工作。