Clemylia commited on
Commit
055800d
·
verified ·
1 Parent(s): cc5bd5e

Upload 2 files

Browse files
Files changed (2) hide show
  1. __init__-1.py +9 -0
  2. core.py +37 -0
__init__-1.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Sitiai - Framework Python pour créer et entraîner des IA légères
3
+ """
4
+
5
+ from .core import create
6
+ from .neuron import SitiNEUR
7
+
8
+ __version__ = "0.1.0"
9
+ __all__ = ["create", "SitiNEUR"]
core.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Module central de Sitiai contenant l'API principale
3
+ """
4
+
5
+ from .generative import GenerativeAI
6
+ from .linear import LinearAI
7
+
8
+
9
+ class CreateAPI:
10
+ """API pour créer différents types d'IA"""
11
+
12
+ def ai(self, ai_type: str, **kwargs):
13
+ """
14
+ Crée une nouvelle IA selon le type spécifié
15
+
16
+ Args:
17
+ ai_type: Type d'IA à créer ('generative' ou 'linear')
18
+ **kwargs: Paramètres additionnels selon le type d'IA
19
+
20
+ Returns:
21
+ Instance d'IA correspondante
22
+
23
+ Example:
24
+ >>> import sitiai
25
+ >>> ai = sitiai.create.ai('generative', mode='name_generator')
26
+ >>> ai = sitiai.create.ai('linear', input_size=10, output_size=1)
27
+ """
28
+ if ai_type == 'generative':
29
+ return GenerativeAI(**kwargs)
30
+ elif ai_type == 'linear':
31
+ return LinearAI(**kwargs)
32
+ else:
33
+ raise ValueError(f"Type d'IA non reconnu: {ai_type}. Utilisez 'generative' ou 'linear'")
34
+
35
+
36
+ # Instance globale de l'API create
37
+ create = CreateAPI()