Spaces:
Sleeping
Sleeping
Siddhanth Sridhar
commited on
Commit
•
33eabc4
1
Parent(s):
a059707
Update app.py
Browse files
app.py
CHANGED
@@ -611,11 +611,11 @@ def main():
|
|
611 |
|
612 |
model.compile(optimizer='adam', loss='mse') # Compile the model
|
613 |
# Explicitly build the model
|
614 |
-
plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
|
615 |
-
img = mpimg.imread('model_architecture.png')
|
616 |
-
plt.imshow(img)
|
617 |
-
plt.axis('off')
|
618 |
-
plt.show()
|
619 |
model.build()
|
620 |
return model
|
621 |
|
@@ -1151,11 +1151,11 @@ def main():
|
|
1151 |
model.compile(optimizer='sgd', loss='mse')
|
1152 |
|
1153 |
model.build() # Explicitly build the model
|
1154 |
-
plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
|
1155 |
-
img = mpimg.imread('model_architecture.png')
|
1156 |
-
plt.imshow(img)
|
1157 |
-
plt.axis('off')
|
1158 |
-
plt.show()
|
1159 |
return model
|
1160 |
|
1161 |
def train_regression_model(df):
|
|
|
611 |
|
612 |
model.compile(optimizer='adam', loss='mse') # Compile the model
|
613 |
# Explicitly build the model
|
614 |
+
# plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
|
615 |
+
# img = mpimg.imread('model_architecture.png')
|
616 |
+
# plt.imshow(img)
|
617 |
+
# plt.axis('off')
|
618 |
+
# plt.show()
|
619 |
model.build()
|
620 |
return model
|
621 |
|
|
|
1151 |
model.compile(optimizer='sgd', loss='mse')
|
1152 |
|
1153 |
model.build() # Explicitly build the model
|
1154 |
+
# plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
|
1155 |
+
# img = mpimg.imread('model_architecture.png')
|
1156 |
+
# plt.imshow(img)
|
1157 |
+
# plt.axis('off')
|
1158 |
+
# plt.show()
|
1159 |
return model
|
1160 |
|
1161 |
def train_regression_model(df):
|