diff --git a/decisionTree.sav b/decisionTree.sav new file mode 100644 index 0000000..0b03640 Binary files /dev/null and b/decisionTree.sav differ diff --git a/injectCode.py b/injectCode.py new file mode 100644 index 0000000..87828bc --- /dev/null +++ b/injectCode.py @@ -0,0 +1,33 @@ +import pickle +import sys + + +def prediction(warzywo, nawoz ,srodek, stan_wzrostu): + filename = 'pliki/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('pliki/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:]) \ No newline at end of file