o
    &Ph)  ã                   @   s*   d dl Z d dlmZ G dd„ dejƒZdS )é    Nc                       s$   e Zd Z‡ fdd„Zdd„ Z‡  ZS )ÚChatbotModelc                    sR   t t| ƒ ¡  t |d¡| _t dd¡| _t d|¡| _t ¡ | _	t 
d¡| _d S )Né€   é@   g      à?)Úsuperr   Ú__init__ÚnnÚLinearÚfc1Úfc2Úfc3ÚReLUÚreluÚDropoutÚdropout)ÚselfÚ
input_sizeÚoutput_size©Ú	__class__© ú"/var/www/finedica/chatbot/model.pyr      s   
zChatbotModel.__init__c                 C   sB   |   |  |¡¡}|  |¡}|   |  |¡¡}|  |¡}|  |¡}|S )N)r   r	   r   r
   r   )r   Úxr   r   r   Úforward   s   


zChatbotModel.forward)Ú__name__Ú
__module__Ú__qualname__r   r   Ú__classcell__r   r   r   r   r      s    r   )ÚtorchÚtorch.nnr   ÚModuler   r   r   r   r   Ú<module>   s    