joanmii/nba-agent-flan-t5

NBA Agent Flan-T5 - Function Calling

Modelo Google Flan-T5 Large especializado en function calling para consultas NBA. Mapea consultas en lenguaje natural a funciones específicas con sus parámetros correspondientes.

🎯 Función Principal

Este modelo recibe una consulta en lenguaje natural sobre la NBA y la convierte en:

• —Nombre de función apropiada (ej: get_player_stats, get_team_info)
• —Parámetros estructurados en formato clave-valor

📝 Formato de Entrada y Salida

Entrada: Consulta en español/inglés

"Estadísticas de Lebron James en la temporada 2023-24"

Salida: Función y parámetros

get_player_stats | player_name='LeBron James', season=2023-24

🔧 Uso

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("joanmii/nba-agent-flan-t5")
model = AutoModelForSeq2SeqLM.from_pretrained("joanmii/nba-agent-flan-t5")

# Consulta de ejemplo
query = "triples dobles de russell westbrook en playoffs"
inputs = tokenizer(query, return_tensors="pt", truncation=True)
outputs = model.generate(
    **inputs, 
    max_new_tokens=64,
    do_sample=False,
    num_beams=1
)
result = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(result)
# Salida: get_triple_doubles | player_name='Russell Westbrook', season_type='Playoffs'

🏀 Funciones Disponibles

El modelo puede mapear consultas a estas 22 funciones NBA:

👤 Funciones de Jugadores

• —get_seasons - Temporadas disponibles de un jugador
• —get_player_info - Información básica de jugadores (altura, peso, equipo, etc.)
• —get_player_stats - Estadísticas detalladas por temporada/carrera
• —get_player_awards - Premios y reconocimientos (MVP, DPOY, etc.)
• —get_player_games - Registro de partidos individuales
• —get_high_low - Máximos y mínimos estadísticos de un jugador
• —get_best_game - Mejores (o peores) partidos de un jugador
• —count_games - Contar partidos con condiciones específicas
• —get_triple_doubles - Triples dobles y dobles-dobles

🏆 Funciones de Liderazgo y Récords

• —get_all_time_leaders - Líderes históricos de la NBA
• —get_league_leaders - Líderes de temporada específica

🏀 Funciones de Equipos

• —get_team_info - Información general de equipos
• —get_franchise_leaders - Líderes históricos por franquicia
• —get_team_roster - Plantillas actuales por temporada
• —get_team_year_by_year_stats - Estadísticas históricas de equipos
• —get_league_standings - Clasificaciones de liga

🎮 Funciones de Partidos

• —get_games - Búsqueda de partidos por criterios
• —get_game - Detalles de partidos específicos
• —get_game_stats - Estadísticas detalladas de un partido

🏆 Funciones de Eventos Especiales

• —get_awards - Premios anuales (MVP, ROY, etc.)
• —get_draft_history - Historial del draft NBA
• —get_nba_champions - Campeones históricos de la NBA
• —get_playoffs - Información de playoffs por año

📊 Ejemplos de Mapeo

🎯 Detalles Técnicos

• —Modelo base: google/flan-t5-large
• —Arquitectura: Transformer seq2seq (encoder-decoder)
• —Tarea: Function calling / Tool use para NBA
• —Formato: función | parámetro1=valor1, parámetro2=valor2
• —Idiomas: Español (principal) e inglés
• —Dominio: NBA, baloncesto profesional

🚀 Integración

Este modelo está diseñado para integrarse con sistemas de herramientas NBA que pueden ejecutar las funciones generadas y devolver los datos solicitados.

💻 Código Fuente

El código completo para usar la aplicación del modelo disponible en GitHub: 👉 joanmii/nba_agent