lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update skipgram embedding example

Browse Source
This commit is contained in:
fchollet 2015-05-08 20:36:17 -07:00
parent 1837458957
commit 03fa41801a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/skipgram_word_embeddings.py
View File

@ -105,8 +105,8 @@ if train_model:
else: else:
print('Build model...') print('Build model...')
model = Sequential() model = Sequential()
model.add(WordContextProduct(max_features, proj_dim=dim_proj, init="normal")) model.add(WordContextProduct(max_features, proj_dim=dim_proj, init="uniform"))
model.compile(loss='hinge', optimizer='adam') model.compile(loss='mse', optimizer='rmsprop')
sampling_table = sequence.make_sampling_table(max_features) sampling_table = sequence.make_sampling_table(max_features)