AI
word2vec 속도 개선 - 개선판 word2vec 학습
CBOW 모델 구현 # CBOW 모델 구현 import sys sys.path.append('..') import numpy as np from layers import Embedding class CBOW: def __init__(self, vocab_size, hidden_size, window_size, corpus): V, H = vocab_size, hidden_size # 가중치 초기화 W_in = 0.01* np.random.rand(V, H).astype('f') W_out = 0.01*np.random.rand(V, H).astype('f') # 계층 생성 self.in_layers = [] for i in range(2 * window_size): layer = Embedding(W_i..
2021. 1. 7. 01:24
최근댓글