Tuberia / Tuberia.py
Thebull's picture
Create Tuberia.py
c791165
raw
history blame
1.27 kB
de escribir importar Dict , Lista
importar numpy como np
clase PreTrainedPipeline ():
def __init__ ( self, ruta = "" )):
# IMPLEMENTAR_ESTO
# Precargue todos los elementos que necesitar谩 en la inferencia.
# Por ejemplo, su modelo, procesadores, tokenizador que podr铆an ser necesarios.
# Esta funci贸n solo se llama una vez, as铆 que haz todo el procesamiento pesado de E/S aqu铆"""
generar NotImplementedError(
"Implemente la funci贸n __init__ PreTrainedPipeline"
)
def __call__ ( self, entradas: str ) -> Lista [ Lista [ Dict [ str , float ]]]:
"""
Argumentos:
entradas (:obj:`str`):
una cadena que contiene algo de texto
Devolver:
A :obj:`lista`:. El objeto devuelto debe ser una lista de una lista como [[{"label": 0.9939950108528137}]] que contenga:
- "etiqueta": una cadena que representa cu谩l es la etiqueta/clase. Puede haber varias etiquetas.
- "puntuaci贸n": una puntuaci贸n entre 0 y 1 que describe la confianza del modelo para esta etiqueta/clase.
"""
# IMPLEMENTAR_ESTO
generar NotImplementedError(
"Implemente la funci贸n __call__ PreTrainedPipeline"