Add MLJ.{un,}supervisedtask methods

JuliaAI · Apr 16, 2019 · f13e0c1 · f13e0c1
1 parent 19ea457
commit f13e0c1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/tasks.jl b/src/tasks.jl
@@ -40,3 +40,8 @@ function coerce(types::Dict{Symbol, Type}, X)
     names = schema(X).names
     NamedTuple{names}(coerce(types[name], selectcols(X, name)) for name in names)
 end
+
+supervisedtask(; data=nothing, types=nothing, kwargs...) =
+    SupervisedTask(; data = coerce(types, data), kwargs...)
+unsupervisedtask(; data=nothing, types=nothing, kwargs...) =
+    UnsupervisedTask(; data = coerce(types, data), kwargs...)