diff --git a/06_extract_features_and_predict.py b/06_extract_features_and_predict.py index 421048a..4a5689e 100644 --- a/06_extract_features_and_predict.py +++ b/06_extract_features_and_predict.py @@ -215,6 +215,6 @@ if __name__ == "__main__": #print("Features extracted from the file {}".format(args.filepath)) #print(pe_features) - saved_model = joblib.load("models/malware_classifier_4.pkl") + saved_model = joblib.load("/home/ubuntu/bigData/projet_big_data/models/malware_classifier_4.pkl") prediction = predict_from_features(pe_features, saved_model) print(prediction) diff --git a/check_files.sh b/07_check_files.sh similarity index 100% rename from check_files.sh rename to 07_check_files.sh diff --git a/check_files2.sh b/08_check_files2.sh similarity index 100% rename from check_files2.sh rename to 08_check_files2.sh