27 lines
730 B
Python
27 lines
730 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]]))
|
||
|
|
print(decision(val))
|
||
|
|
|
||
|
|
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:])
|