PigeonAIModel1 / script /generate_csv.py
tapigeon's picture
Enriched train Data
372555a
raw
history blame contribute delete
586 Bytes
import os
import pandas as pd
import random
bad_words = ["stupid", "idiot", "hate you", "shut up", "worst", "trash", "dumb"]
good_words = ["hello", "thank you", "nice", "amazing", "respect", "kind", "helpful"]
data=[]
for _ in range(500):
data.append({"text": random.choice(bad_words), "label": 1}) #bad
for _ in range(500):
data.append({"text": random.choice(good_words), "label": 0}) #good
random.shuffle(data)
df = pd.DataFrame(data)
# ✅ Create 'data' directory if it doesn't exist
os.makedirs("./../data", exist_ok=True)
df.to_csv("./../data/text.csv", index=False)