2017-10-07 95 views
0

我有這個類,它構建的分類功能,在文本求得所有載體的詞矢量使用numpy python數組打印類輸出?

class MeanEmbeddingVectorizer(object): 
    def __init__(self, word2vec): 
     self.word2vec = word2vec 
     self.dim = len(word2vec.itervalues().next()) 

    def fit(self, X, y): 
     return self 

    def transform(self, X): 
     return np.array([ 
      np.mean([self.word2vec[w] for w in words if w in self.word2vec] 
        or [np.zeros(self.dim)], axis=0) 
      for words in X 
     ]) 

所以我的問題是如何打印這個類的輸出到屏幕或將其保存到一個文件 謝謝

回答

1

你可以在末尾添加一個print聲明,但你仍然需要創建屬性word2vec

import numpy as np 

class MeanEmbeddingVectorizer(object): 
    def __init__(self, word2vec): 
     self.word2vec = word2vec 
     self.dim = len(word2vec.itervalues().next()) 

    def fit(self, X, y): 
     return self 

    def transform(self, X): 
     return np.array([ 
      np.mean([self.word2vec[w] for w in words if w in self.word2vec] 
        or [np.zeros(self.dim)], axis=0) 
      for words in X 
     ]) 
print(MeanEmbeddingVectorizer.fit("X", "Y", "Z")) 

會給你的輸出

X 
None 

你要是跑不過

print(MeanEmbeddingVectorizer.transform("X", "Y")) 

AttributeError: 'str' object has no attribute 'word2vec'