In [1]:

    

import numpy
from physlearn.NeuralNet.NeuralNet import NeuralNet
import tensorflow as tf
import matplotlib.pyplot as plt
%matplotlib inline


    

In [2]:

    

x = numpy.array([[-2, -1, 0, 1, 2]])
y = numpy.array([[0, 0, 0, 1, 1]])


    

In [3]:

    

net = NeuralNet(-1, 1)


    

In [4]:

    

net.add_input_layer(1)
net.add(5, tf.sigmoid)
net.add_output_layer(1, tf.sigmoid)


    

In [5]:

    

net.compile()


    

In [6]:

    

net.run(x)


    

    
Out[6]:





array([[ 0.22197409,  0.20184572,  0.18965699,  0.18605342,  0.18741254]])

In [7]:

    

net.calculate_cost(x, y)


    

    
Out[7]:





0.28975827693245554

In [8]:

    

net.set_train_type('logistic')


    

In [9]:

    

cost_list = net.train(x, y, 5, 1000, 1)


    

    




100%|██████████| 1000/1000 [00:00<00:00, 1786.91it/s]

In [10]:

    

plt.plot(cost_list)


    

    
Out[10]:





[<matplotlib.lines.Line2D at 0x7fc99c883358>]






    

In [11]:

    

net.run(x)


    

    
Out[11]:





array([[ 0.03414385,  0.0512385 ,  0.21153253,  0.80103775,  0.93231733]])

In [ ]: