File size: 986 Bytes
9274888 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | 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
|