bring back "log_vector"

Author: nomadbl

src/Loggers/LogHistograms.jl

@@ -31,6 +31,17 @@ function log_histogram(logger::TBLogger, name::AbstractString, data::AbstractArr
     write_event(logger.file, make_event(logger, summ, step=step))
 end
 
+"""
+    log_vector(logger, name, data::Vector; step=step(logger))
+Logs the vector found in `data` as an histogram under the name `name`.
+"""
+function log_vector(logger::TBLogger, name::AbstractString, data::AbstractVector; step=nothing)
+    # @warn "TensorBoardLogger Depracation Warning: log_vector does not correctly log vectors. Use bar diagrams instead using a plotting library."
+    hist = Histogram(collect(0:length(data)), data)
+    summ = SummaryCollection(histogram_summary(name, hist))
+    write_event(logger.file, make_event(logger, summ, step=step))
+end
+
 function histogram_summary(name::AbstractString, hist::Histogram{T,1}) where T
     edges     = first(hist.edges)
     hist_vals = hist.weights

src/TensorBoardLogger.jl

@@ -20,7 +20,7 @@ using  Base.CoreLogging: CoreLogging, AbstractLogger, LogLevel, Info,
     NullLogger
 
 export TBLogger, reset!, set_step!, increment_step!, set_step_increment!
-export log_histogram, log_value, log_text, log_image, log_images,
+export log_histogram, log_value, log_vector, log_text, log_image, log_images,
        log_audio, log_audios, log_graph, log_embeddings, log_custom_scalar
 export map_summaries, TBReader
 

test/runtests.jl

@@ -84,6 +84,7 @@ end
         log_histogram(logger, "hist/cust", data_tuple, step=step)
         log_histogram(logger, "hist/cust", rand(100), step=step)
         log_histogram(logger, "hist/cust", rand(10,10), step=step)
+        log_vector(logger, "hist/cust", rand(10), step=step)
 
         close.(values(logger.all_files))
     end