Juliabelloni/weather-forecasting
0
1�
2 ��f� � �@ � d dl Zd dl mZ d dlmZ G d� d� � ZdS )� N)�keras)�layersc �< � e Zd Zd� Zddededefd�Zdd3ed efd �Zd
S )�NNc � � d S )N� )�selfs �E/Users/juliabelloni/Desktop/ML_industry/ML4IProject/src/models/fnn.py�__init__z NN.__init__ s � � �� �adam� �N� � optimizer�units�shapec �� � t j t j |�� � t j |d�� � t j dd�� � t j d� � g� � }|� |d�� � |S )N)r �relu��4activation� � �mean_squared_error)r �loss)r �5Sequentialr �Input�Dense�compile)r r r r �models r6 �fnnzNN.fnn7 sv � �� �� �5�)�)�)�� �U�v�6�6�6�� �R�F�3�3�3�� �Q��� "8� � �� �
�
� �0D�
�E�E�E�� r �9 �h_units� input_shapec �� � t j t j ||d�� � t j dd�� � g� � }|� ddt j � � � g�� � || _ |S ) N�tanh)r r$ r r �linearr r r
)r r �metrics) r r r � SimpleRNNr r r( �RootMeanSquaredError�_rnn)r r# r$ r s r10 �rnnzNN.rnn s� � �� � � �7�{�f� V� V� V� �L��h� /� /� /� 11� � �� �
�
�/�6� �}�9�9�;�;�<� � >� >� >� �� �� r N)r
r r )r" r ) �__name__�12__module__� __qualname__r �str�int�tupler! r, r r r13 r r sx � � � � � �
�
�
�� �C� �� �%� � � � � � �3� �%� � � � � � r r )�14tensorflow�tfr �tensorflow.kerasr r r r r15 �<module>r6 sg �� � � � � � � � � � � #� #� #� #� #� #�� � � � � � � � � r