Skip to content

Commit

Permalink
Merge pull request #44 from Evovest/dev
Browse files Browse the repository at this point in the history
Dev
  • Loading branch information
jeremiedb authored Feb 20, 2020
2 parents 85009a0 + 1519e9c commit 090dbcd
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 1 deletion.
2 changes: 1 addition & 1 deletion Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "EvoTrees"
uuid = "f6006082-12f8-11e9-0c9c-0d5d367ab1e5"
authors = ["jeremiedb <[email protected]>"]
version = "0.4.4"
version = "0.4.5"

[deps]
CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"
Expand Down
5 changes: 5 additions & 0 deletions src/MLJ.jl
Original file line number Diff line number Diff line change
Expand Up @@ -69,30 +69,35 @@ MLJModelInterface.metadata_pkg.((EvoTreeRegressor, EvoTreeClassifier, EvoTreeCou
uuid="f6006082-12f8-11e9-0c9c-0d5d367ab1e5",
url="https://github.com/Evovest/EvoTrees.jl",
julia=true,
license="Apache",
is_wrapper=false)

MLJModelInterface.metadata_model(EvoTreeRegressor,
input=MLJModelInterface.Table(MLJModelInterface.Continuous),
target=AbstractVector{<:MLJModelInterface.Continuous},
weights=false,
path="EvoTrees.EvoTreeRegressor",
descr=EvoTreeRegressor_desc)

MLJModelInterface.metadata_model(EvoTreeClassifier,
input=MLJModelInterface.Table(MLJModelInterface.Continuous),
target=AbstractVector{<:MLJModelInterface.Finite},
weights=false,
path="EvoTrees.EvoTreeClassifier",
descr=EvoTreeClassifier_desc)

MLJModelInterface.metadata_model(EvoTreeCount,
input=MLJModelInterface.Table(MLJModelInterface.Continuous),
target=AbstractVector{<:MLJModelInterface.Count},
weights=false,
path="EvoTrees.EvoTreeCount",
descr=EvoTreeCount_desc)

MLJModelInterface.metadata_model(EvoTreeGaussian,
input=MLJModelInterface.Table(MLJModelInterface.Continuous),
target=AbstractVector{<:MLJModelInterface.Continuous},
weights=false,
path="EvoTrees.EvoTreeGaussian",
descr=EvoTreeGaussian_desc)

# shared metadata
Expand Down

0 comments on commit 090dbcd

Please sign in to comment.