[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-07-15 08:31:27 +03:00
parent ed0afe832d
commit 916540ad08

@ -301,7 +301,7 @@ for word, i in tqdm(word_index.items()):
embedding_matrix[i] = embedding_vector embedding_matrix[i] = embedding_vector
# %% [code] # %% [code]
%%time #%%time
with strategy.scope(): with strategy.scope():
# A simple LSTM with glove embeddings and one dense layer # A simple LSTM with glove embeddings and one dense layer
@ -355,7 +355,7 @@ scores_model.append({'Model': 'LSTM','AUC_Score': roc_auc(scores,yvalid)})
# ## Code Implementation # ## Code Implementation
# %% [code] # %% [code]
%%time #%%time
with strategy.scope(): with strategy.scope():
# GRU with glove embeddings and two dense layers # GRU with glove embeddings and two dense layers
model = Sequential() model = Sequential()
@ -397,7 +397,7 @@ scores_model
# ## Code Implementation # ## Code Implementation
# %% [code] # %% [code]
%%time #%%time
with strategy.scope(): with strategy.scope():
# A simple bidirectional LSTM with glove embeddings and one dense layer # A simple bidirectional LSTM with glove embeddings and one dense layer
model = Sequential() model = Sequential()