Modelpublic

HCKLab/BiBert-MultiTask-1

sourceHugging Facemitupdated 4y agoView on Hugging Face

0likes7downloads

root__pycache__bibert_multitask_classification.cpython-37.pyc

bibert_multitask_classification.cpython-37.pyc10 linesDownload Raw Back to __pycache__

1B

2�Vc�@s8ddlmZddlZddlZdd�ZGdd�de�ZdS)�)�PipelineNcCs0tj|ddd�}t�||�}||jddd�S)N�����T)�axis�keepdims)�np�max�exp�sum)�_outputs�maxes�shifted_exp�r
�>/content/BiBert-MultiTask-1/bibert_multitask_classification.py�softmaxsrc@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)3�BiBert_MultiTaskPipelinecKs:i}d|kr|d|d<i}d|kr0|d|d<||ifS)N�task_idr
)�self�kwargsZpreprocess_kwargsZforward_kwargsr
r
r�_sanitize_parameterssz-BiBert_MultiTaskPipeline._sanitize_parameterscCs@|j}|j|d|d��|j�}tjd|td�}t�|�|d<|S)NT)�padding�return_tensors�)�shape�4fill_value�dtype�task_ids)	�	framework�	tokenizer�to�devicer�full�int�torch�	IntTensor)r�inputsrr�featurerr
r
r�5preprocesss6z#BiBert_MultiTaskPipeline.preprocesscCs|jf|�S)N)�model)r�model_inputsrr
r
r�_forwardsz!BiBert_MultiTaskPipeline._forwardcs6|dd}|��}t|�}�fdd�t|�D�}|S)N�logitsrcs(g|] \}}�jjj||��d��qS))�label�score)r'�config�id2label�item)�.0�ir,)rr
r�7<listcomp>'sz8BiBert_MultiTaskPipeline.postprocess.<locals>.<listcomp>)�numpyr�	enumerate)r�
model_outputs�outputs�scores�dict_scoresr
)rr�postprocess!s8z$BiBert_MultiTaskPipeline.postprocessN)�__name__�9__module__�__qualname__rr&r)r9r
r
r
rr10sr)�transformersrr3rr"rrr
r
r
r�<module>s