Skip to content

Commit 254cd06

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent bbb02ec commit 254cd06

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

explain.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,9 @@ def update_sample(samples, N, sample):
4646
)
4747

4848
# Load the trained model
49-
model = tf.keras.models.load_model(os.path.join(args.input_dir, "nn_model_sherlock.keras"))
49+
model = tf.keras.models.load_model(
50+
os.path.join(args.input_dir, "nn_model_sherlock.keras")
51+
)
5052

5153
# Produce a randomly sample of background from the training data
5254
background = []

train.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -51,31 +51,31 @@
5151
regex_model1 = BatchNormalization(axis=1)(regex_model_input)
5252
regex_model2 = Dense(
5353
1000,
54-
activation='relu',
54+
activation="relu",
5555
kernel_regularizer=tf.keras.regularizers.l2(0.0001),
5656
)(regex_model1)
5757
regex_model3 = Dropout(0.35)(regex_model2)
5858
regex_model4 = Dense(
5959
1000,
60-
activation='relu',
60+
activation="relu",
6161
kernel_regularizer=tf.keras.regularizers.l2(0.0001),
6262
)(regex_model3)
6363

6464
merged_model2 = BatchNormalization(axis=1)(regex_model4)
6565
merged_model3 = Dense(
6666
500,
67-
activation='relu',
67+
activation="relu",
6868
kernel_regularizer=tf.keras.regularizers.l2(0.0001),
6969
)(merged_model2)
7070
merged_model4 = Dropout(0.35)(merged_model3)
7171
merged_model5 = Dense(
7272
500,
73-
activation='relu',
73+
activation="relu",
7474
kernel_regularizer=tf.keras.regularizers.l2(0.0001),
7575
)(merged_model4)
7676
merged_model_output = Dense(
7777
len(le.classes_),
78-
activation='softmax',
78+
activation="softmax",
7979
kernel_regularizer=tf.keras.regularizers.l2(0.0001),
8080
)(merged_model5)
8181

0 commit comments

Comments
 (0)