Update .travis.yml

.travis.yml

@@ -11,6 +11,4 @@ script:
 addons:
   apt:
     packages:
-     - python3.6
-     - python3-pip
      - graphviz

tests/test_ann.py

@@ -0,0 +1,24 @@
+import keras;
+from keras.models import Sequential;
+from keras.layers import Dense;
+
+network = Sequential();
+        #Hidden Layer#1
+network.add(Dense(units=6,
+                  activation='relu',
+                  kernel_initializer='uniform',
+                  input_dim=11));
+
+        #Hidden Layer#2
+network.add(Dense(units=6,
+                  activation='relu',
+                  kernel_initializer='uniform'));
+
+        #Exit Layer
+network.add(Dense(units=1,
+                  activation='sigmoid',
+                  kernel_initializer='uniform'));
+
+from ann_visualizer.visualize import ann_viz;
+
+ann_viz(network, title="");