Skip to content

Instantly share code, notes, and snippets.

@hussius
hussius / decode_cossmo_example.py
Created April 26, 2018 11:02 — forked from hannes-brt/decode_cossmo_example.py
Function to decode a COSSMO training example in tfrecord format
def read_single_cossmo_example(serialized_example, n_tissues=1, coord_sys='rna1'):
"""Decode a single COSSMO example
coord_sys must be one of 'rna1' or 'dna0', if 'dna0' then an extra 'strand' field
must exist in the tfrecord and is extracted.
"""
assert coord_sys in ['dna0', 'rna1']
context_features = {