| defined_as: kipoi.model.PyTorchModel | |
| args: | |
| module_file: model_files/pyt.py | |
| module_obj: simple_model | |
| weights: model_files/only_weights.pth | |
| default_dataloader: . # path to the directory | |
| info: | |
| authors: | |
| - name: Roman Kreuzhuber | |
| github: krrome | |
| doc: Simple testing model for pytorch | |
| dependencies: | |
| conda: | |
| - pytorch::pytorch>=0.2.0 | |
| schema: | |
| inputs: | |
| name: input | |
| shape: (1000, 4) | |
| special_type: DNASeq | |
| doc: DNA sequence | |
| # associated_metadata: ranges # --> has to be defined in dataloader.yaml. | |
| # This field is ignored in model.yaml. | |
| targets: | |
| shape: (1, ) | |
| doc: Predicted binding strength | |
| column_labels: | |
| - some_probability | |
| test: | |
| expect: | |
| url: https://s3.eu-central-1.amazonaws.com/kipoi-models/predictions/example/models/pyt/expected.pred.h5 | |
| md5: d6d0779a7bdfb1301c76a59defd293ed | |
| precision_decimal: 6 | |