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

Consulta	Función + Parámetros
"MVP de Stephen Curry"	get_player_awards | player_name='Stephen Curry', award_type='MVP'
"Roster de Lakers 2024"	get_team_roster | team_name='Los Angeles Lakers', season='2024-25'
"Estadísticas de Michael Jordan"	get_player_stats | player_name='Michael Jordan'
"Triples dobles de Westbrook playoffs"	get_triple_doubles | player_name='Russell Westbrook', season_type='Playoffs'
"Mejores anotadores 2023"	get_league_leaders | stat='PTS', season='2023-24'
"Draft de 2003"	get_draft_history | season='2003'
"Campeones NBA 2016"	get_nba_champions | year=2016
"Partidos Lakers vs Warriors"	get_games | team1='Los Angeles Lakers', team2='Golden State Warriors'
"Líderes históricos en rebotes"	get_all_time_leaders | stat='REB'
"Playoffs 2024"	get_playoffs | year=2024

🎯 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