animetix-web / src /scripts /run_self_play_debate.py
MissawB's picture
Upload folder using huggingface_hub
6a8fee8 verified
Raw
History Blame Contribute Delete
1.66 kB
import os
import argparse
from backend.animetix.services import AnimetixService
def run_self_play_session(limit: int = 5):
"""
Lance une session automatique de Self-Play Debating pour générer
des données de raisonnement complexe pour le DPO/Fine-tuning.
"""
print("⚔️ Starting Self-Play Debating Session...")
animetix = AnimetixService()
debate_service = animetix.self_play_debate_service
catalog = animetix.load_data('Anime')
if not catalog:
print("❌ Catalog not found.")
return
sample_items = catalog['db'][:limit]
topics = [
"Le développement du personnage principal est mal rythmé.",
"La fin de l'œuvre détruit tout le message précédent.",
"Les thèmes philosophiques sont superficiels et prétentieux."
]
import random
success_count = 0
for item in sample_items:
title = item.get('title', 'Inconnu')
topic = random.choice(topics)
try:
record = debate_service.run_debate(target_media=title, topic=topic)
if record:
success_count += 1
except Exception as e:
print(f"❌ Debate failed for {title}: {e}")
print("\n" + "="*40)
print(f"🏁 Session completed. {success_count}/{limit} debates recorded.")
print(f"💾 Check '{debate_service.dataset_path}' for training data.")
print("="*40)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--limit", type=int, default=3)
args = parser.parse_args()
run_self_play_session(limit=args.limit)