mirror of
https://github.com/Ladebeze66/llm_lab_perso.git
synced 2025-12-15 19:16:51 +01:00
20 lines
574 B
Python
20 lines
574 B
Python
# Tester un seul agent, avec des paramètres qui varient
|
|
import sys
|
|
import os
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
from core.factory import LLMFactory
|
|
from utils.parameter_tester import test_agents_on_prompt
|
|
|
|
test_agents_on_prompt(
|
|
model_class=LLMFactory._registry["mistral7b"],
|
|
prompt="Donne une définition claire de l'intelligence artificielle.",
|
|
agents=["juriste"],
|
|
param_grid={
|
|
"temperature": [0.1, 0.5, 1.0],
|
|
"top_p": [0.7, 0.9]
|
|
},
|
|
fixed_params={
|
|
"num_predict": 256
|
|
}
|
|
)
|