From 1058dfaa179b381a30cf90585496cb24c7e9b12e Mon Sep 17 00:00:00 2001
From: Elias Projahn <elias@johrpan.de>
Date: Mon, 22 Nov 2021 13:32:50 +0100
Subject: [PATCH] neural: Add progress information

---
 R/neural.R | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/R/neural.R b/R/neural.R
index 5c9e8d2..91f332c 100644
--- a/R/neural.R
+++ b/R/neural.R
@@ -8,6 +8,9 @@ neural <- function(preset, progress = NULL, seed = 49641) {
         set.seed(seed)
         gene_count <- length(gene_ids)
 
+        progress_buffer <- 0
+        progress_step <- 1 / (2 * length(reference_gene_ids) + 1)
+
         # Prefilter distances by species.
         distances <- geposan::distances[species %chin% species_ids]
 
@@ -146,6 +149,11 @@ neural <- function(preset, progress = NULL, seed = 49641) {
             data_matrix <- keras::normalize(data_matrix)
 
             data[gene %chin% gene_ids, score := predict(model, data_matrix)]
+
+            if (!is.null(progress)) {
+                progress_buffer <<- progress_buffer + progress_step
+                progress(progress_buffer)
+            }
         }
 
         # Apply the network to all non-training genes first.