Skip to content

Commit 2d54b7c

Browse files
committed
initiated mlj interface port
1 parent 9c61a5a commit 2d54b7c

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

Project.toml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ authors = ["Bernard Brenyah", "Andrey Oskin"]
44
version = "0.1.0"
55

66
[deps]
7+
MLJModelInterface = "e80e1ace-859a-464e-9ed9-23947d8ae3ea"
78
StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
89

910
[compat]
@@ -12,8 +13,9 @@ julia = "1.3"
1213

1314
[extras]
1415
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
15-
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
1616
Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
17+
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
18+
MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d"
1719

1820
[targets]
19-
test = ["Test", "Random", "Suppressor"]
21+
test = ["Test", "Random", "Suppressor", "MLJBase"]

src/mlj_interface.jl

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# TODO 1: a using MLJModelInterface or import MLJModelInterface statement
2+
using MLJModelInterface
3+
using ParallelKMeans
4+
5+
# TODO 2: MLJ-compatible model types and constructors,
6+
7+
# TODO 3: implementation of fit, predict/transform and optionally fitted_params for your models,
8+
9+
# TODO 4: metadata for your package and for each of your models
10+
11+
12+
13+

0 commit comments

Comments
 (0)