Skip to main content

Una libreria simple para crear modelos de lenguaje con tensorflow

Project description

Una libreria simple para crear modelos de lenguaje de forma practica y sencilla usando tensorflow en pocas lineas de codigo

instalacion de la libreria:

pip install cuellartensorflow

Creando nuestro primer modelo: el modelo se entrena con datos directos de wikipedia por lo que no es necesario descargar ningun dataset.

from cuellartensorflow import train_language_model

modelo = train_language_model(
    language="es",
    num_articles=10,
    num_epochs=10
)

cambia el parametro "es" de la funcion languaje="es" para entrenar el modelo con un idioma distinto consulta la documentacion de wikipedia para mayor informacion, "num_articles" para especificar el numero de articulos a usar para el entrenamiento, "num_epochs" para especificar el numero de epocas a usar para el entrenamiento del modelo

Para generar texto usando nuestro modelo recien creado puedes usar el siguiente codigo:

from cuellartensorflow import generate_text

texto=generate_text(
    seed_text="La educacion",
    num_words=15,
    temperature=0.7
)

print(texto)

usa el apartado "seed_text" para especificar la palabra semilla para generar texto a partir de ella, "num_words" sirve para especificar el numero de palabras a generar por el modelo "temperature" controla la temperatura para la creatividad del modelo

la libreria es muy facil de usar para cualquier principiante que quiera crear su propio modelo de lenguaje de forma rapida y sencilla

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

cuellartensorflow-1.0.2.tar.gz (3.7 kB view hashes)

Uploaded Source

Built Distribution

cuellartensorflow-1.0.2-py3-none-any.whl (4.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page