Add haslistrules isensemble

Author: giopaglia

src/SoleModels.jl

@@ -62,7 +62,9 @@ export DecisionForest, trees
 
 export MixedModel
 
-export solemodel
+export haslistrules, solemodel
+
+export isensemble
 
 include("types/model.jl")
 include("types/AbstractTrees.jl")

src/rule-extraction.jl

@@ -42,8 +42,13 @@ See also [`listrules`](@ref), [`Rule`](@ref)], [`issymbolicmodel`](@ref).
 """
 struct PlainRuleExtractor <: RuleExtractor end
 isexact(::PlainRuleExtractor) = true
-extractrules(::PlainRuleExtractor, m, args...; kwargs...) = listrules(m, args...; kwargs...)
-
+function extractrules(::PlainRuleExtractor, m, args...; kwargs...)
+    if haslistrules(m)
+        listrules(m, args...; kwargs...)
+    else
+        listrules(solemodel(m), args...; kwargs...)
+    end
+end
 
 ############################################################################################
 ############################################################################################

src/types/api.jl

@@ -1,10 +1,20 @@
+# TODO document, together with issymbolic and listrules
+"""
+    haslistrules(m::Any)
+
+This function extracts symbolic final rules from a symbolic model..
+
+See also [`AbstractModel`](@ref), [`listrules`](@ref)
+[`LeafModel`](@ref).
+"""
+haslistrules(m) = false
+haslistrules(m::AbstractModel) = true
 
 # TODO document, together with issymbolic and listrules
 """
     solemodel(m::Any)
 
 This function translates a symbolic model to a symbolic model using the structures defined in SoleModel.jl.
-# Interface
 
 See also [`AbstractModel`](@ref), [`ConstantModel`](@ref), [`FunctionModel`](@ref),
 [`LeafModel`](@ref).
@@ -58,3 +68,7 @@ natoms(m::AbstractModel) = error("Please, provide method natoms(::$(typeof(m))).
 nconnectives(m::AbstractModel) = error("Please, provide method nconnectives(::$(typeof(m))).")
 """$doc_syntax_utils_models"""
 nsyntaxleaves(m::AbstractModel) = error("Please, provide method nsyntaxleaves(::$(typeof(m))).")
+
+
+# TODO
+isensemble(m) = false

src/utils/models/ensembles.jl

@@ -98,6 +98,9 @@ struct DecisionEnsemble{O,T<:AbstractModel,A<:Base.Callable,W<:Union{Nothing,Abs
 
 end
 
+
+isensemble(m::DecisionEnsemble) = true
+
 modelstype(m::DecisionEnsemble{O,T}) where {O,T} = T
 models(m::DecisionEnsemble) = m.models
 nmodels(m::DecisionEnsemble) = length(models(m))
@@ -314,6 +317,8 @@ struct MaxDecisionBag{O,TO<:AbstractModel,TU<:AbstractModel
     end
 end
 
+isensemble(m::MaxDecisionBag) = true
+
 function apply(m::MaxDecisionBag, d::AbstractInterpretation; suppress_parity_warning = false, kwargs...)
     weights = [apply(wm, d; suppress_parity_warning, kwargs...) for wm in m.weight_producing_models]
     om = m.output_producing_models[argmax(weights)]