Mercurial > repos > bgruening > nn_classifier
comparison nn_classifier.xml @ 20:fa36c40c2990 draft
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit f54ff2ba2f8e7542d68966ce5a6b17d7f624ac48
author | bgruening |
---|---|
date | Fri, 13 Jul 2018 03:50:09 -0400 |
parents | c64f57fe1b97 |
children |
comparison
equal
deleted
inserted
replaced
19:73195a06ff03 | 20:fa36c40c2990 |
---|---|
23 | 23 |
24 @COLUMNS_FUNCTION@ | 24 @COLUMNS_FUNCTION@ |
25 @GET_X_y_FUNCTION@ | 25 @GET_X_y_FUNCTION@ |
26 | 26 |
27 input_json_path = sys.argv[1] | 27 input_json_path = sys.argv[1] |
28 params = json.load(open(input_json_path, "r")) | 28 with open(input_json_path, "r") as param_handler: |
29 | 29 params = json.load(param_handler) |
30 | 30 |
31 #if $selected_tasks.selected_task == "load": | 31 #if $selected_tasks.selected_task == "load": |
32 | 32 |
33 classifier_object = pickle.load(open("$infile_model", 'r')) | 33 with open("$infile_model", 'rb') as model_handler: |
34 classifier_object = pickle.load(model_handler) | |
34 | 35 |
35 header = 'infer' if params["selected_tasks"]["header"] else None | 36 header = 'infer' if params["selected_tasks"]["header"] else None |
36 data = pandas.read_csv("$selected_tasks.infile_data", sep='\t', header=header, index_col=None, parse_dates=True, encoding=None, tupleize_cols=False) | 37 data = pandas.read_csv("$selected_tasks.infile_data", sep='\t', header=header, index_col=None, parse_dates=True, encoding=None, tupleize_cols=False) |
37 prediction = classifier_object.predict(data) | 38 prediction = classifier_object.predict(data) |
38 prediction_df = pandas.DataFrame(prediction) | 39 prediction_df = pandas.DataFrame(prediction) |
56 | 57 |
57 my_class = getattr(sklearn.neighbors, classifier) | 58 my_class = getattr(sklearn.neighbors, classifier) |
58 classifier_object = my_class(**options) | 59 classifier_object = my_class(**options) |
59 classifier_object.fit(X, y) | 60 classifier_object.fit(X, y) |
60 | 61 |
61 pickle.dump(classifier_object,open("$outfile_fit", 'w+')) | 62 with open("$outfile_fit", 'wb') as out_handler: |
63 pickle.dump(classifier_object, out_handler) | |
62 | 64 |
63 #end if | 65 #end if |
64 | 66 |
65 ]]> | 67 ]]> |
66 </configfile> | 68 </configfile> |