33 lines
745 B
Python
33 lines
745 B
Python
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:]) |