added alternate dataset

This commit is contained in:
Kamil Ryżek 2023-05-26 02:10:21 +02:00
parent 0c096589d7
commit 255e45fd93

View File

@ -0,0 +1,33 @@
import csv
import random
battery_values = ['high', 'medium', 'low']
distance_values = ['far', 'medium', 'close']
mood_values = ['good', 'medium', 'bad']
other = ['yes', 'no']
training_data = []
for _ in range(200):
battery = random.choice(battery_values)
distance = random.choice(distance_values)
mood = random.choice(mood_values)
memory = random.randint(0,4)
dishes_held = random.randint(0,4)
"""empty_basket = random.choice(other)
if empty_basket == 'yes':
dish_in_basket = 'no'
else:
dish_in_basket = random.choice(other)
dish_in_basket = random.choice(other)"""
waiting_for_order = random.choice(other)
waiting_for_dish = random.choice(other)
example = [battery, distance, mood, memory, dishes_held, waiting_for_order, waiting_for_dish]
training_data.append(example)
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(training_data)