Implement Tables.jl interface for LabeledTable

Author: juliohm

src/StatsLearnModels.jl

@@ -4,13 +4,13 @@
 
 module StatsLearnModels
 
-using Tables
 using Distances
 using DataScienceTraits
 using StatsBase: mode, mean
 using ColumnSelectors: ColumnSelector, selector
 using TableTransforms: StatelessFeatureTransform
 
+import Tables
 import TableTransforms: applyfeat, isrevertible
 
 using DecisionTree: AdaBoostStumpClassifier, DecisionTreeClassifier, RandomForestClassifier

src/labeledtable.jl

@@ -23,6 +23,18 @@ function LabeledTable(table, names)
   LabeledTable{typeof(table)}(table, labs)
 end
 
+Tables.istable(::Type{<:LabeledTable}) = true
+
+Tables.rowaccess(::Type{<:LabeledTable{T}}) where {T} = Tables.rowaccess(T)
+
+Tables.columnaccess(::Type{<:LabeledTable{T}}) where {T} = Tables.columnaccess(T)
+
+Tables.rows(t::LabeledTable) = Tables.rows(t.table)
+
+Tables.columns(t::LabeledTable) = Tables.columns(t.table)
+
+Tables.columnnames(t::LabeledTable) = Tables.columnnames(t.table)
+
 """
     label(table, names)