Skip to content
This repository has been archived by the owner on Nov 28, 2023. It is now read-only.

DeepRank output format #247

Open
manonreau opened this issue Feb 4, 2022 · 0 comments
Open

DeepRank output format #247

manonreau opened this issue Feb 4, 2022 · 0 comments

Comments

@manonreau
Copy link
Contributor

Describe the bug
DeepRank output format is not user friendly

Environment:
Not relevant

To Reproduce
See any example from https://github.com/DeepRank/deeprank/tree/master/paper_pretrained_models

Expected Results
The output of the network is an HDF5 file that contains the raw network prediction (no softmax transformation for classification tasks) which is hard to understand for someone unfamiliar with NN. Extracting a score or a rank from the default output is not straightforward.

It would be nice to add:

  • for classification task: apply a softmax transformation, and eventually save the raw data, the transformed ones, and the predicted class
  • a hdf5-to-csv convertor to easily extract the information from the hdf5 output files and provide a human-readable format

This is already implemented in DeepRank-GNN and the hdf5 to csv convertor can be easily adapted (see https://github.com/DeepRank/Deeprank-GNN/blob/master/deeprank_gnn/tools/hdf5_to_csv.py)

Actual Results or Error Info

Additional Context

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

1 participant