stage v0.1.7

Author: levimcclenny

setup.py

@@ -13,7 +13,7 @@
 
 setuptools.setup(
     name="tensordiffeq",
-    version="0.1.6.7",
+    version="0.1.7",
     author="Levi McClenny",
     author_email="[email protected]",
     description="Distributed PDE Solver in Tensorflow",

tensordiffeq/output.py

@@ -2,6 +2,8 @@
 from os import system, name
 import sys
 
-def print_screen(Domain, model):
+def print_screen(model):
     f = Figlet(font='slant')
-    print(f.renderText('TensorDiffEq'))
+    print(f.renderText('TensorDiffEq'))
+    print("Neural Network Model Summary\n")
+    print(model.u_model.summary())