File size: 1,085 Bytes
055800d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | """
Module central de Sitiai contenant l'API principale
"""
from .generative import GenerativeAI
from .linear import LinearAI
class CreateAPI:
"""API pour créer différents types d'IA"""
def ai(self, ai_type: str, **kwargs):
"""
Crée une nouvelle IA selon le type spécifié
Args:
ai_type: Type d'IA à créer ('generative' ou 'linear')
**kwargs: Paramètres additionnels selon le type d'IA
Returns:
Instance d'IA correspondante
Example:
>>> import sitiai
>>> ai = sitiai.create.ai('generative', mode='name_generator')
>>> ai = sitiai.create.ai('linear', input_size=10, output_size=1)
"""
if ai_type == 'generative':
return GenerativeAI(**kwargs)
elif ai_type == 'linear':
return LinearAI(**kwargs)
else:
raise ValueError(f"Type d'IA non reconnu: {ai_type}. Utilisez 'generative' ou 'linear'")
# Instance globale de l'API create
create = CreateAPI()
|