Fix position weight matrix assignment
This commit is contained in:
parent
02d20d4e72
commit
035162bd8d
|
@ -22,10 +22,10 @@ def generate_example(sequence, reference_sequence, weight_matrix) -> bytes:
|
|||
"reference_sequence": Feature(
|
||||
int64_list=Int64List(value=list(encode_sequence(reference_sequence)))
|
||||
),
|
||||
"A_counts": Feature(float_list=FloatList(value=[weight_matrix["A"][0]])),
|
||||
"C_counts": Feature(float_list=FloatList(value=[weight_matrix["C"][0]])),
|
||||
"G_counts": Feature(float_list=FloatList(value=[weight_matrix["G"][0]])),
|
||||
"T_counts": Feature(float_list=FloatList(value=[weight_matrix["T"][0]])),
|
||||
"A_counts": Feature(float_list=FloatList(value=weight_matrix["A"])),
|
||||
"C_counts": Feature(float_list=FloatList(value=weight_matrix["C"])),
|
||||
"G_counts": Feature(float_list=FloatList(value=weight_matrix["G"])),
|
||||
"T_counts": Feature(float_list=FloatList(value=weight_matrix["T"])),
|
||||
}
|
||||
example = Example(features=Features(feature=schema))
|
||||
return example.SerializeToString()
|
||||
|
|
Loading…
Reference in New Issue