billusanda007 commited on
Commit
d728717
·
verified ·
1 Parent(s): 2b16bf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -66,12 +66,12 @@ def load_deeprank_model(max_sequence_length):
66
 
67
  model = Sequential()
68
 
69
- #model.add(Embedding(input_dim=vocab_size, output_dim=128, input_length=max_sequence_length))
70
- #model.add(Conv1D(filters=128, kernel_size=5, activation='relu'))
71
- #model.add(MaxPooling1D(pool_size=2))
72
- #model.add(LSTM(64))
73
- #model.add(Dense(num_classes, activation='softmax'))
74
- #model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
75
  model.load_weights('deeprank_model_v2.h5')
76
  return model
77
 
 
66
 
67
  model = Sequential()
68
 
69
+ model.add(Embedding(input_dim=vocab_size, output_dim=128, input_length=max_sequence_length))
70
+ model.add(Conv1D(filters=128, kernel_size=5, activation='relu'))
71
+ model.add(MaxPooling1D(pool_size=2))
72
+ model.add(LSTM(64))
73
+ model.add(Dense(num_classes, activation='softmax'))
74
+ model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
75
  model.load_weights('deeprank_model_v2.h5')
76
  return model
77