johrpan/geposan
1
0
Fork 0
mirror of https://github.com/johrpan/geposan.git synced 2025-10-26 18:57:25 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

neural: Use optimizer adam

Browse source
This commit is contained in:
Elias Projahn 2021-12-05 19:05:57 +01:00
parent 8ced026b79
commit 397340cc64
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
R/neural.R
View file

@ -6,7 +6,7 @@
# training data and validated using this set. For non-training genes, the
# final score will be the mean of the result of applying the different
# models.
neural <- function(preset, progress = NULL, seed = 49641, n_models = 5) {
neural <- function(preset, progress = NULL, seed = 751833, n_models = 5) {
species_ids <- preset$species_ids
gene_ids <- preset$gene_ids
reference_gene_ids <- preset$reference_gene_ids
@ -175,7 +175,10 @@ neural <- function(preset, progress = NULL, seed = 49641, n_models = 5) {
units = 1,
activation = "sigmoid"
) |>
keras::compile(loss = "binary_crossentropy")
keras::compile(
loss = keras::loss_binary_crossentropy(),
optimizer = keras::optimizer_adam()
)
# Train the model.