Mercurial > repos > ufz > dfpl_train
comparison json_predict.py @ 0:e0bb949eac45 draft default tip
planemo upload for repository https://github.com/Helmholtz-UFZ/galaxy-tools/tree/main/tools/dfpl commit 66c6acfeff5441c36fba97787ddc5ee3d6a4a6ec
author | ufz |
---|---|
date | Thu, 19 Dec 2024 12:51:21 +0000 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:e0bb949eac45 |
---|---|
1 import json | |
2 from sys import stdin | |
3 | |
4 d = json.load(stdin) | |
5 | |
6 # The directory where the full model of the fnn is loaded from. | |
7 d["fnnModelDir"] = "" # 'dfpl predict' looks for "model_weights.h5" in this directory | |
8 del d["fnn_weights"] | |
9 | |
10 d["compressFeatures"] = bool(d["compressFeatures"] == "true") | |
11 | |
12 # The encoder file where it is loaded from, to compress the fingerprints. | |
13 d["ecModelDir"] = "" | |
14 d["ecWeightsFile"] = "encoder_weights.h5" | |
15 | |
16 # Output csv file name which will contain one prediction per input line. | |
17 # Default: prefix of input file name. | |
18 d["outputFile"] = "predictions.csv" | |
19 | |
20 d["py/object"] = "dfpl.options.Options" | |
21 | |
22 print(json.dumps(d)) |