From 5ac81c049faab888b1639525605016503ac18a04 Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Tue, 1 Jun 2021 18:34:29 +0200
Subject: [PATCH] Change BASES constant to a local variable
---
src/model.py | 6 ++----
src/preprocessing.py | 2 --
2 files changed, 2 insertions(+), 6 deletions(-)
diff --git a/src/model.py b/src/model.py
index de6e87c..dbf3bbe 100644
--- a/src/model.py
+++ b/src/model.py
@@ -1,10 +1,8 @@
from tensorflow.keras import Model, Sequential, layers
from tensorflow.keras.regularizers import l2
-from preprocessing import BASES
-
-def build_model(hyper_parameters) -> Model:
+def build_model(hyper_parameters, bases="ACGT") -> Model:
"""
Builds the CNN model
"""
@@ -41,6 +39,6 @@ def build_model(hyper_parameters) -> Model:
),
layers.Dropout(rate=0.3),
# Output layer with softmax activation
- layers.Dense(units=len(BASES), activation="softmax"),
+ layers.Dense(units=len(bases), activation="softmax"),
]
)
diff --git a/src/preprocessing.py b/src/preprocessing.py
index f04b6d6..93fa5b6 100644
--- a/src/preprocessing.py
+++ b/src/preprocessing.py
@@ -4,8 +4,6 @@ from numpy.random import random
from tensorflow.io import TFRecordWriter
from tensorflow.train import BytesList, Example, Feature, Features, FloatList
-BASES = "ACGT"
-
def generate_example(sequence, weight_matrix):
schema = {