ostateczna wersja

This commit is contained in:
Tomasz Dzierzbicki 2020-06-09 17:12:24 +00:00
parent 5978098747
commit bedbdbf1e2
3 changed files with 903 additions and 424 deletions

1192
Main.cpp

File diff suppressed because it is too large Load Diff

BIN
decisionTree.sav Normal file

Binary file not shown.

33
injectCode.py Normal file
View File

@ -0,0 +1,33 @@
import pickle
import sys
def prediction(warzywo, nawoz ,srodek, stan_wzrostu):
filename = 'decisionTree.sav'
tree = pickle.load(open(filename, 'rb'))
val = (tree.predict([[warzywo, nawoz, srodek, stan_wzrostu]]))
save(val)
def save(prediction):
pred = str(prediction)
plik = open('dec.txt', 'w')
plik.write(pred[1])
plik.close()
def decision(prediction):
if prediction == 0:
return "Nie_podejmuj_dzialania"
elif prediction == 1:
return "Zastosuj_nawoz"
elif prediction == 2:
return "Zastosuj_srodek"
elif prediction == 4:
return "Zbierz"
elif prediction == 5:
return "Roslina_juz_zgnila__zbierz_i_wyrzuc"
if __name__ == '__main__':
# Map command line arguments to function arguments.
prediction(*sys.argv[1:])