SI_Traktor/dataset.py

40 lines
1.5 KiB
Python

header = ['pogoda', 'ile_chwastow', 'ile_burakow', 'czy_chce_pracowac']
training_data = [
['slonecznie', 'duzo', 'malo', 'tak'],
['slonecznie', 'srednio', 'srdenio', 'tak'],
['slonecznie', 'bardzo_duzo', 'malo', 'srednio'],
['slonecznie', 'malo', 'bardzo_duzo', 'tak'],
['slonecznie', 'duzo', 'bardzo_malo', 'srednio'],
['slonecznie', 'malo', 'bardzo_malo', 'srednio'],
['deszcz', 'bardzo_duzo', 'malo', 'sredino'],
['deszcz', 'srednio', 'srednio', 'srednio'],
['deszcz', 'malo', 'bardzo_duzo', 'tak'],
['deszcz', 'duzo', 'duzo', 'srednio'],
['deszcz', 'malo', 'malo', 'tak'],
['deszcz', 'srednio', 'bardzo_duzo', 'tak'],
['grad', 'srednio', 'malo', 'srednio'],
['grad', 'bradzo_malo', 'bardzo_duzo', 'srednio'],
['grad', 'duzo', 'srednio', 'srednio'],
['grad', 'malo', 'bardzo_malo', 'srednio'],
['grad', 'bardzo_duzo', 'bardzo_malo', 'nie'],
['grad', 'srednio', 'bardzo_malo', 'nie'],
['zachmurzenie', 'srednio', 'srednio', 'tak'],
['zachmurzenie', 'bardzo_duzo', 'duzo', 'srednio'],
['zachmurzenie', 'malo', 'srednio', 'tak'],
['zachmurzenie', 'duzo', 'malo', 'srednio'],
['zachmurzenie', 'malo', 'bardzo_malo', 'tak'],
['zachmurzenie', 'malo', 'srednio', 'tak']
]
testing_data = [
['slonecznie', 'srednio', 'srednio', 'tak'],
['deszcz', 'duzo', 'malo', 'nie'],
['deszcz', 'duzo', 'duzo', 'srednio'],
['grad', 'srednio', 'duzo', 'nie'],
['zachmurzenie', 'bardzo_duzo', 'malo', 'nie'],
['zachmurzenie', 'bardzo_duzo', 'malo', 'nie']
]