Update app.py
Browse files
app.py
CHANGED
@@ -133,13 +133,14 @@ async def train_and_save_model():
|
|
133 |
if os.path.exists('chatbot_model.h5'):
|
134 |
model = load_model('chatbot_model.h5')
|
135 |
else:
|
136 |
-
|
137 |
-
|
138 |
-
|
139 |
-
|
140 |
-
|
141 |
-
|
142 |
-
|
|
|
143 |
|
144 |
sgd = SGD(learning_rate=0.01, momentum=0.9, nesterov=True)
|
145 |
model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['accuracy'])
|
|
|
133 |
if os.path.exists('chatbot_model.h5'):
|
134 |
model = load_model('chatbot_model.h5')
|
135 |
else:
|
136 |
+
input_layer = Input(shape=(len(train_x[0]),))
|
137 |
+
layer1 = Dense(128, activation='relu')(input_layer)
|
138 |
+
layer2 = Dropout(0.5)(layer1)
|
139 |
+
layer3 = Dense(64, activation='relu')(layer2)
|
140 |
+
layer4 = Dropout(0.5)(layer3)
|
141 |
+
output_layer = Dense(len(classes), activation='softmax')(layer4)
|
142 |
+
|
143 |
+
model = Sequential(layers=[input_layer, layer1, layer2, layer3, layer4, output_layer])
|
144 |
|
145 |
sgd = SGD(learning_rate=0.01, momentum=0.9, nesterov=True)
|
146 |
model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['accuracy'])
|