Merge pull request #18 from JuliaAI/new-traits

Add new traits to to support better MLJ machine type checks and composite models

Author: ablaom

Project.toml

@@ -1,7 +1,7 @@
 name = "StatisticalTraits"
 uuid = "64bff920-2084-43da-a3e6-9bb72801c0c9"
 authors = ["Anthony D. Blaom <[email protected]>"]
-version = "2.0.1"
+version = "2.1.0"
 
 [deps]
 ScientificTypesBase = "30f210dd-8aff-4c5f-94ba-8e64358c1161"

src/StatisticalTraits.jl

@@ -9,6 +9,10 @@ const TRAITS = [
     :input_scitype,
     :output_scitype,
     :target_scitype,
+    :fit_data_scitype, #
+    :predict_scitype,  #
+    :transform_scitype, #
+    :inverse_transform_scitype, #
     :is_pure_julia,
     :package_name,
     :package_license,
@@ -24,6 +28,7 @@ const TRAITS = [
     :human_name,
     :is_supervised,
     :prediction_type,
+    :abstract_type, #
     :hyperparameters,
     :hyperparameter_types,
     :hyperparameter_ranges,
@@ -133,6 +138,10 @@ snakecase(s::Symbol) = Symbol(snakecase(string(s)))
 input_scitype(::Type)          = Unknown
 output_scitype(::Type)         = Unknown
 target_scitype(::Type)         = Unknown
+fit_data_scitype(::Type)       = Unknown
+predict_scitype(::Type)        = Unknown
+transform_scitype(::Type)      = Unknown
+inverse_transform_scitype(::Type)  = Unknown
 
 # The following refer to properties of the package defining a type,
 # for use in, say, a registry of machine learning models. All but the
@@ -157,6 +166,7 @@ prediction_type(::Type)        = :unknown
 
 # Miscellaneous:
 
+abstract_type(::Type)   = Any
 is_wrapper(::Type)      = false     # or `true`
 supports_online(::Type) = false     # or `true`
 docstring(M::Type)      = string(M) # some `String`