2016-12-15 12 views
7

से वजन matrices प्राप्त करें मैं अजगर में gensim word2vec पैकेज का उपयोग कर रहा हूँ। मैं W और W' वजन मैट्रिस को पुनर्प्राप्त करना चाहता हूं जो स्किप-ग्राम सीखने के दौरान सीख चुके हैं।gensim word2Vec

ऐसा लगता है कि model.syn0 मुझे पहला प्रदान करता है लेकिन मुझे यकीन नहीं है कि मैं दूसरा कैसे प्राप्त कर सकता हूं। कोई उपाय?

मैं वास्तव में मॉडल सुलभ विशेषताओं पर किसी भी संपूर्ण दस्तावेज़ प्राप्त करने के क्योंकि अधिकारी एक (के लिए उदाहरण syn0 एक विशेषता के रूप में वर्णित नहीं है) सटीक होना करने के लिए

+1

मैंने अभी एक छोटे डेटासेट के साथ प्रयास किया और यह मुझे पूरे एम्बेडिंग मैट्रिक्स 'np.shape (model.syn0) दे रहा है। –

उत्तर

2

प्रतीत नहीं होता है प्यार model.wv.syn0इनपुट एम्बेडिंग शामिल हैं मैट्रिक्स। आउटपुट एम्बेडिंग model.syn1 में संग्रहीत किया जाता है, जब यह hierarchical softmax (hs=1) के साथ प्रशिक्षित है या model.syn1neg में जब यह नकारात्मक नमूना (negative>0) का उपयोग करता है। बस! जब दोनों पदानुक्रमित सॉफ्टमैक्स और नकारात्मक नमूना सक्षम नहीं होते हैं, तो Word2Vec प्रशिक्षण के लिए एकल वजन मैट्रिक्स model.wv.syn0 का उपयोग करता है।

एक संबंधित चर्चा here भी देखें।

संबंधित मुद्दे