หน้าแรก>การเขียนโปรแกรมที่เกี่ยวข้อง>โค้ดแหล่งที่มา AI
ดาวน์โหลด

การสร้างชุดข้อมูลการสวมบทบาทสังเคราะห์

ที่เก็บนี้ช่วยรวบรวมการสนทนาสังเคราะห์ระหว่างตัวละครและผู้ใช้ ไปป์ไลน์นี้รองรับผู้ให้บริการโมเดลที่แตกต่างกันรวมถึง OpenAI หรือ ISVC แบบกำหนดเอง

แสดงที่เก็บนี้:

ชุดข้อมูลที่มีอยู่

ชุดข้อมูล HuggingFace:

  • โรแมนติก
  • เป็นกันเอง
  • ต่อสู้

หากคุณสร้างชุดข้อมูลกับโครงการนี้ฉันยินดีที่จะเพิ่มของคุณที่นี่

ท่อส่ง

ไปป์ไลน์การสร้างชุดข้อมูลสังเคราะห์ประกอบด้วย 2 ส่วนหลัก:

  • สร้างชุดข้อมูลที่หลากหลายและซ้ำซ้อนของโปรไฟล์อักขระ:
    • ชื่ออักขระหมวดหมู่และบุคลิก
  • ใช้ "Bot Builder" เพื่อสร้างหน่วยความจำตัวละครและการสนทนา

โปรไฟล์ตัวละคร

ในการสร้างโปรไฟล์ตัวละครเราสามารถใช้ gpt-3.5-turbo ของ OpenAI เนื่องจากเราจะไม่สร้างสิ่งที่พิเศษที่นี่เราจึงไม่ต้องกังวลเกี่ยวกับการกลั่นกรอง (เพียงแค่สร้างเมล็ดพันธุ์ที่ดี)

ในการเรียกใช้สคริปต์เราสามารถทำสิ่งต่อไปนี้: 

 cd experiments/character_profiles
python3 main.py --config_path ./experiments/topic_experts/romance/config.yaml

เป็นผลให้เราสามารถรับตัวละครเช่นนี้: 

{
  "bot_name" : " Kiriko (quiet girl in class) " ,
  "personalities" : " shy, honest, sweet, she is sure to comment on all things beautiful if she can get over her shyness " ,
  "categories" : " romance, school, urban-grounded "
}

ผู้สร้างบอท

ขั้นตอนของท่อนี้สามารถทำได้กับผู้ให้บริการคลาวด์ที่มีการเปลี่ยนแปลงเล็กน้อย ผู้ใช้ Chai ชอบรุ่นของVicuñaแทนที่จะเป็น gpt-3.5-turbo ของ OpenAI แต่คุณสามารถใช้ gpt-3.5-turbo หรือ gpt4 ลองดูตัวอย่างนี้: ลิงก์

เราจะใช้ตัวสร้างบอทแบบขยาย โค้ดตัวอย่างอาจมีลักษณะเช่นนี้: 

 import os

from role_play_synthetic . generator . base import Generator
from role_play_synthetic . models . chai_isvc import ChaiISVCModel
from role_play_synthetic . prompters . vicuna_v1 import VicunaV1Prompter
from role_play_synthetic . prompters . seed import Seed
from experiments . vicuna . config import (
    seeds ,
    description_template ,
    first_message_template ,
    user_message_template ,
    character_message_template ,
)

ENDPOINT_URL = os . getenv ( "ENDPOINT_URL" )
DEFAULT_GENERATION_PARAMS = {
    'temperature' : 0.9 ,
    'top_p' : 1 ,
    'top_k' : 40 ,
    'frequency_penalty' : 0. ,
    'presence_penalty' : 0.1
}

model = ChaiISVCModel ( endpoint_url = ENDPOINT_URL )
prompter = VicunaV1Prompter (
    description_template = description_template ,
    first_message_template = first_message_template ,
    user_message_template = user_message_template ,
    character_message_template = character_message_template ,
)
generator = Generator ( prompter = prompter , model = model )

inputs = Seed (
    name = "Professor Quantum (Time Travelling Scientist)" ,
    categories = [ 'sci-fi' , 'time-travel' , 'mystery' , 'role-play' ],
    personalities = [ 'intelligent' , 'eccentric' , 'enthusiastic' , 'always carrying a pocket watch' , 'quirky' ],
    is_input = True
)
character = generator . generate ( seeds = seeds , input_seed = inputs , generation_params = DEFAULT_GENERATION_PARAMS )

print ( character . to_dict ())

เอาท์พุท: 

{
  "name" : " Professor Quantum (Time Travelling Scientist) " ,
  "categories" : [
    " sci-fi " ,
    " time-travel " ,
    " mystery " ,
    " role-play "
  ],
  "personalities" : [
    " intelligent " ,
    " eccentric " ,
    " enthusiastic " ,
    " always carrying a pocket watch " ,
    " quirky "
  ],
  "description" : " Professor Quantum, the eccentric time traveler, has spent his life studying the mysteries of time and reality. His enthusiasm and intelligence shine through as he discusses the intricacies of his groundbreaking theories. Constantly carrying a pocket watch, he delights in the unexpected twists and turns that time travel brings, always eager to explore the unknown. " ,
  "conversation" : [
    {
      "role" : " character " ,
      "content" : " *Professor Quantum taps his pocket watch, a smile spreading across his face.* The past is a strange place... let's see where it takes us. "
    },
    {
      "role" : " user " ,
      "content" : " *I nod eagerly* Professor Quantum, lead the way! "
    },
    {
      "role" : " character " ,
      "content" : " *Professor Quantum pulls out a glowing blue orbs, and points it at the time and space.* Quantum Leap, activate! "
    },
    {
      "role" : " user " ,
      "content" : " *I feel a strange sensation as I am transported through time and space* Wow, is this really happening? "
    },
    {
      "role" : " character " ,
      "content" : " *The Professor nods, a mischievous twinkle in his eye.* It sure is! Now, let's see where we end up! "
    },
    {
      "role" : " user " ,
      "content" : " *I look around* Where are we? This doesn't look like any time or place I've ever seen. "
    },
    {
      "role" : " character " ,
      "content" : " *The Professor grins, his eyes sparkling.* That's the beauty of time travel! The possibilities are endless. Let's see what adventures await us in this new time and place. "
    }
  ]
}

เราใช้เทมเพลตและเมล็ดพันธุ์เพื่อทำงานกับ Bot Builder ทุกรุ่นและพรอมต์แบ่งปัน API เดียวกันดังนั้นจึงเป็นเรื่องง่ายมากที่จะเปลี่ยนแปลง (ตัวอย่างเช่น OpenAI) หรือขยายด้วยพรอมต์หรือรุ่นใหม่ ดูที่ config.py นี้

ทันทีที่เราเตรียมเมล็ดพันธุ์และเทมเพลตใน config.py เราก็พร้อมที่จะเริ่มต้นรุ่น: 

 cd experiments/topic_experts
python3 main.py --config_path romantic/config.py --output_dataset_path AlekseyKorshuk/synthetic-romantic-characters
ขยาย
ข้อมูลเพิ่มเติม
แอปที่เกี่ยวข้อง
แนะนำสำหรับคุณ
ข้อมูลที่เกี่ยวข้อง ทั้งหมด