Merge pull request #7 from JuliaML/show

Update FittedModel show

Author: juliohm

src/interface.jl

@@ -29,3 +29,5 @@ struct FittedModel{M,C}
   model::M
   cache::C
 end
+
+Base.show(io::IO, ::FittedModel{M}) where {M} = print(io, "FittedModel{$(nameof(M))}")

test/runtests.jl

@@ -17,7 +17,13 @@ const SLM = StatsLearnModels
   output = iris[:, [:target]]
   train, test = MLJ.partition(1:nrow(input), 0.7, rng=123)
 
-  @testset "interface" begin
+  @testset "show" begin
+    model = DecisionTreeClassifier()
+    fmodel = SLM.fit(model, input[train, :], output[train, :])
+    @test sprint(show, fmodel) == "FittedModel{DecisionTreeClassifier}"
+  end
+
+  @testset "models" begin
     @testset "MLJ" begin
       Random.seed!(123)
       Tree = MLJ.@load(DecisionTreeClassifier, pkg = DecisionTree, verbosity = 0)