Merge pull request #4 from alan-turing-institute/iteration

Add iteration_parameter and supports_training_losses

Project.toml

@@ -1,7 +1,7 @@
 name = "StatisticalTraits"
 uuid = "64bff920-2084-43da-a3e6-9bb72801c0c9"
 authors = ["Anthony D. Blaom <[email protected]>"]
-version = "0.1.0"
+version = "0.1.1"
 
 [deps]
 ScientificTypes = "321657f4-b219-11e9-178b-2701a2544e81"

src/StatisticalTraits.jl

@@ -31,7 +31,9 @@ const TRAITS = [
     :reports_each_observation,
     :aggregation,
     :is_feature_dependent,
-    :distribution_type
+    :distribution_type,
+    :iteration_parameter,
+    :supports_training_losses
 ]
 
 
@@ -150,7 +152,7 @@ supports_class_weights(::Type) = false
 # Possible values of the following are `:deterministic`, `:probabilistic`,
 # `:interval`, or `:unknown`:
 
-prediction_type(::Type)        = :unknown 
+prediction_type(::Type)        = :unknown
 
 # Miscellaneous:
 
@@ -164,6 +166,8 @@ aggregation(::Type)     = Mean()
 is_feature_dependent(::Type)     = false
 reports_each_observation(::Type) = false
 distribution_type(::Type)        = missing
+iteration_parameter(::Type)      = nothing
+supports_training_losses(::Type) = false
 
 # Returns a tuple, with one entry per field of `T` (the type of some
 # statistical model, for example). Each entry is `nothing` or defines