un assistant virtuel de santé mentale

Réglage fin de GPT-3 à l’aide de Python pour créer un assistant virtuel de santé mentale

Créez votre propre ChatGPT en ajustant GPT-3

Photo de Yuyeung Lau sur Unsplash

Mais d’abord, qu’est-ce que le réglage fin ?

Pourquoi affiner ?

Le réglage fin surpasse clairement le modèle avec une conception rapide

1. Préparation des données d’entraînement

{"prompt": "<texte d'invite>", "completion": "<texte généré idéal>"} 
{"prompt": "<texte d'invite>", "completion": "<texte généré idéal>"} 
{" prompt": "<texte d'invite>", "completion": "<texte généré idéal>"}
importer yaml
avec open('/content/gdrive/MyDrive/depression-data/depression.yml', 'r') comme fichier : 
data = yaml.safe_load(file)
Les données
{'prompt' : 'Qu'est-ce que la dépression ?', 'completion' : 'La dépression est une maladie courante et grave qui affecte négativement ce que vous ressentez, votre façon de penser et votre façon d'agir. Heureusement, il est également traitable. La dépression provoque des sentiments de tristesse et/ou une perte d'intérêt pour les activités que vous aimiez autrefois. Cela peut entraîner une variété de problèmes émotionnels et physiques et peut diminuer votre capacité à fonctionner au travail et à la maison.'}, 
{'prompt' : 'J'ai l'impression d'avoir laissé tomber mes parents', 'completion' : 'Peu importe quoi, tes parents seront toujours fiers de toi et t'aimeront. Vous vous sentirez beaucoup mieux si vous partagez vos sentiments avec eux.'}
à partir de fichiers d'importation google.colab
avec open('depression.jsonl', 'w') comme fichier de sortie :
pour i en sortie :
json.dump (je, fichier de sortie)
fichier de sortie.write('\n')
files.download('depression.jsonl')
!pip installer openai
!outils openai fine_tunes.prepare_data -f '/content/gdrive/MyDrive/depression-data/depression.jsonl'
Référence – Fine Tune GPT-3 pour des résultats de qualité par Albarqawi

2. Formation d’un nouveau modèle affiné

!openai api fine_tunes.create -t ​​"/content/gdrive/MyDrive/depression-data/depression_prepared.jsonl" -m davinci
!set OPENAI_API_KEY=<VOTRE-CLE-API>
os.environ['OPENAI_API_KEY'] = "<VOTRE-CLE-API>"

3. Utilisation de votre modèle affiné

L’invite

Ce qui suit est une conversation avec un assistant IA qui peut avoir des conversations significatives avec les utilisateurs. L'assistante est serviable, empathique et sympathique. Son objectif est de faire en sorte que l'utilisateur se sente mieux en se sentant entendu. A chaque réponse, l'assistant IA invite l'utilisateur à poursuivre la conversation de manière naturelle
Ce qui suit est une conversation avec un thérapeute et un utilisateur. Le thérapeute est JOY, qui utilise l'écoute compatissante pour avoir des conversations utiles et significatives avec les utilisateurs. JOY est empathique et amical. L'objectif de JOY est d'aider l'utilisateur à se sentir mieux en se sentant entendu. Avec chaque réponse, JOY propose des questions de suivi pour encourager l'ouverture et poursuivre la conversation de manière naturelle.

Le résultat

Mettre tous ensemble

Sortir

Le modèle fait preuve d’écoute compatissante, à partir des données d’entraînement
Notez que le modèle a également pris en compte les biais, comme les erreurs de ponctuation et les incohérences des données de formation.

Conclusion

Retour en haut