Deep Learning Workshop

@@ -87,12 +87,12 @@ if __name__ == "__main__":
     for sequence, label in zip(sequences, labels):
         if label == "exon":
             count_exon += 1
-            total_exon_length += total_exon_length
+            total_exon_length += len(sequence)
             if len(sequence) > max_exon_length:
                 max_exon_length = len(sequence)
         elif label == "utr":
             count_utr += 1
-            total_utr_length += total_utr_length
+            total_utr_length += len(sequence)
             if len(sequence) > max_utr_length:
                 max_utr_length = len(sequence)
     print("Max exon length: {}".format(max_exon_length))
@@ -101,6 +101,8 @@ if __name__ == "__main__":
     print("Count utr: {}".format(count_utr))
     print("Total exon length: {}".format(total_exon_length))
     print("Total utr length: {}".format(total_utr_length))
+    print("Media exon length: {}".format(total_exon_length/count_exon))
+    print("Media utr length: {}".format(total_utr_length/count_utr))
 
     quit()