Add IO methods for LabeledTable

Author: juliohm

Project.toml

@@ -1,7 +1,7 @@
 name = "StatsLearnModels"
 uuid = "c146b59d-1589-421c-8e09-a22e554fd05c"
-authors = ["Elias Carvalho <[email protected]> and contributors"]
 version = "1.1.1"
+authors = ["Elias Carvalho <[email protected]> and contributors"]
 
 [deps]
 ColumnSelectors = "9cc86067-7e36-4c61-b350-1ac9833d277f"
@@ -11,7 +11,9 @@ Distances = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 GLM = "38e38edf-8417-5370-95a0-9cbb8c7f171a"
 NearestNeighbors = "b8a86587-4115-5ab1-83bc-aa920d37bbce"
+PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
+StyledStrings = "f489334b-da3d-4c2e-b8f0-e476e12c162b"
 TableTransforms = "0d432bfd-3ee1-4ac1-886a-39f05cc69a3e"
 Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 
@@ -30,7 +32,9 @@ Distributions = "0.25"
 GLM = "1.9"
 MLJModelInterface = "1.9"
 NearestNeighbors = "0.4"
+PrettyTables = "3.0.2"
 StatsBase = "0.33, 0.34"
+StyledStrings = "1.0"
 TableTransforms = "1.15"
 Tables = "1.11"
 julia = "1.9"

src/StatsLearnModels.jl

@@ -5,6 +5,8 @@
 module StatsLearnModels
 
 using Distances
+using PrettyTables
+using StyledStrings
 using DataScienceTraits
 using StatsBase: mode, mean
 using ColumnSelectors: ColumnSelector, selector

src/labeledtable.jl

@@ -35,6 +35,54 @@ Tables.columns(t::LabeledTable) = Tables.columns(t.table)
 
 Tables.columnnames(t::LabeledTable) = Tables.columnnames(t.table)
 
+# -----------
+# IO METHODS
+# -----------
+
+function Base.summary(io::IO, t::LabeledTable)
+  name = nameof(typeof(t))
+  nlab =  length(t.labels)
+  print(io, "$name with $nlab label(s)")
+end
+
+Base.show(io::IO, t::LabeledTable) = summary(io, t)
+
+function Base.show(io::IO, ::MIME"text/plain", t::LabeledTable)
+  pretty_table(io, t; backend=:text, _common_kwargs(t)...)
+end
+
+function Base.show(io::IO, ::MIME"text/html", t::LabeledTable)
+  pretty_table(
+    io,
+    t;
+    backend=:html,
+    _common_kwargs(t)...,
+    renderer=:show,
+    style=HtmlTableStyle(title=["font-size" => "14px"])
+  )
+end
+
+function _common_kwargs(t)
+  cols = Tables.columns(t)
+  vars = Tables.columnnames(cols)
+
+  labels = map(vars) do var
+    if var ∈ t.labels
+      styled"{(weight=bold),magenta:$var}"
+    else
+      styled"{(weight=bold):$var}"
+    end
+  end
+
+  (
+    title=summary(t),
+    column_labels=collect(labels),
+    maximum_number_of_rows=10,
+    new_line_at_end=false,
+    alignment=:c
+  )
+end
+
 """
     label(table, names)