Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Lucas Miranda
deepOF
Commits
48bf4bac
Commit
48bf4bac
authored
Apr 19, 2021
by
lucas_miranda
Browse files
Added extra branch to main autoencoder for rule_based prediction
parent
f0902fce
Pipeline
#99909
passed with stages
in 19 minutes and 35 seconds
Changes
2
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
deepof/train_utils.py
View file @
48bf4bac
...
...
@@ -100,7 +100,7 @@ def get_callbacks(
elif
reg_cat_clusters
and
reg_cluster_variance
:
latreg
=
"categorical+variance"
run_ID
=
"{}{}{}{}{}{}{}{}{}{}{}{}"
.
format
(
run_ID
=
"{}{}{}{}{}{}{}{}{}{}{}{}
{}
"
.
format
(
(
"GMVAE"
if
variational
else
"AE"
),
(
"_input_type={}"
.
format
(
input_type
)
if
input_type
else
"coords"
),
(
"_window_size={}"
,
format
(
X_train
.
shape
[
1
])),
...
...
@@ -116,8 +116,6 @@ def get_callbacks(
(
"_{}"
.
format
(
datetime
.
now
().
strftime
(
"%Y%m%d-%H%M%S"
))
if
not
run
else
""
),
)
print
(
run_ID
)
log_dir
=
os
.
path
.
abspath
(
os
.
path
.
join
(
outpath
,
"fit"
,
run_ID
))
tensorboard_callback
=
tf
.
keras
.
callbacks
.
TensorBoard
(
log_dir
=
log_dir
,
...
...
tests/test_train_utils.py
View file @
48bf4bac
...
...
@@ -33,7 +33,7 @@ def test_load_treatments():
@
given
(
X_train
=
arrays
(
shape
=
st
.
tuples
(
st
.
integers
(
min_value
=
1
,
max_value
=
1000
)),
shape
=
st
.
tuples
(
st
.
integers
(
min_value
=
1
,
max_value
=
1000
)
,
st
.
just
(
24
)
),
dtype
=
float
,
elements
=
st
.
floats
(
min_value
=
0.0
,
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment