Move hardcorded data to a constants module

This commit is contained in:
2021-06-01 19:27:10 +02:00
parent 44ff69dc9e
commit 220c0482f1
3 changed files with 17 additions and 7 deletions

View File

@@ -1,8 +1,10 @@
from tensorflow.keras import Model, Sequential, layers
from tensorflow.keras.regularizers import l2
from constants import BASES
def build_model(hyper_parameters, bases="ACGT") -> Model:
def build_model(hyper_parameters) -> Model:
"""
Builds the CNN model
"""
@@ -39,6 +41,6 @@ def build_model(hyper_parameters, bases="ACGT") -> 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"),
]
)