Skip to content

Commit 2f26e79

Browse files
committed
remove "Abstract" from detector types
1 parent 38091f3 commit 2f26e79

File tree

3 files changed

+16
-38
lines changed

3 files changed

+16
-38
lines changed

src/MLJModelInterface.jl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -44,10 +44,10 @@ const ABSTRACT_MODEL_SUBTYPES =
4444
:UnsupervisedAnnotator,
4545
:SupervisedDetector,
4646
:UnsupervisedDetector,
47-
:AbstractProbabilisticSupervisedDetector,
48-
:AbstractProbabilisticUnsupervisedDetector,
49-
:AbstractDeterministicSupervisedDetector,
50-
:AbstractDeterministicUnsupervisedDetector]
47+
:ProbabilisticSupervisedDetector,
48+
:ProbabilisticUnsupervisedDetector,
49+
:DeterministicSupervisedDetector,
50+
:DeterministicUnsupervisedDetector]
5151

5252

5353
# ------------------------------------------------------------------------
@@ -147,11 +147,11 @@ abstract type UnsupervisedAnnotator <: Annotator end
147147
abstract type UnsupervisedDetector <: UnsupervisedAnnotator end
148148
abstract type SupervisedDetector <: SupervisedAnnotator end
149149

150-
abstract type AbstractProbabilisticSupervisedDetector <: SupervisedDetector end
151-
abstract type AbstractProbabilisticUnsupervisedDetector <: UnsupervisedDetector end
150+
abstract type ProbabilisticSupervisedDetector <: SupervisedDetector end
151+
abstract type ProbabilisticUnsupervisedDetector <: UnsupervisedDetector end
152152

153-
abstract type AbstractDeterministicSupervisedDetector <: SupervisedDetector end
154-
abstract type AbstractDeterministicUnsupervisedDetector <: UnsupervisedDetector end
153+
abstract type DeterministicSupervisedDetector <: SupervisedDetector end
154+
abstract type DeterministicUnsupervisedDetector <: UnsupervisedDetector end
155155

156156
# ------------------------------------------------------------------------
157157
# includes

src/model_api.jl

Lines changed: 2 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@ fit(::Static, ::Integer, data...) = (nothing, nothing, nothing)
1616
fit(m::Supervised, verbosity, X, y, w) = fit(m, verbosity, X, y)
1717

1818
# fallback for unsupervised detectors when no "evaluation" labels appear:
19-
fit(m::Union{AbstractProbabilisticUnsupervisedDetector,
20-
AbstractDeterministicUnsupervisedDetector},
19+
fit(m::Union{ProbabilisticUnsupervisedDetector,
20+
DeterministicUnsupervisedDetector},
2121
verbosity,
2222
X,
2323
y) = fit(m, verbosity, X)
@@ -154,27 +154,6 @@ function transform end
154154
"""
155155
function inverse_transform end
156156

157-
"""
158-
augmented_transform
159-
160-
If implemented, the same as `transform`, but with a return value
161-
augmented by the `transform`ation of the training data.
162-
163-
For example, for if implemented for a `Supervised` model with a
164-
`transform` method, the return value of `transform_transform(model,
165-
fitresult, Xnew)` coincides with
166-
167-
```julia
168-
(transform(model, fitresult, X), transform(model, fitresult, Xnew))
169-
```
170-
171-
where `(X, y)` was the training data.
172-
173-
Must be implemented by any `UnsupervisedDetector` or `SupervisedDetector`.
174-
175-
"""
176-
function augmented_transform end
177-
178157
# models can optionally overload these for enable serialization in a
179158
# custom format:
180159
function save end
@@ -184,4 +163,3 @@ function restore end
184163
some meta-models may choose to implement the `evaluate` operations
185164
"""
186165
function evaluate end
187-

src/model_traits.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22

33
# unexported aliases:
44
const Detector = Union{SupervisedDetector,UnsupervisedDetector}
5-
const ProbabilisticDetector = Union{AbstractProbabilisticSupervisedDetector,
6-
AbstractProbabilisticUnsupervisedDetector}
7-
const DeterministicDetector = Union{AbstractDeterministicSupervisedDetector,
8-
AbstractDeterministicUnsupervisedDetector}
5+
const ProbabilisticDetector = Union{ProbabilisticSupervisedDetector,
6+
ProbabilisticUnsupervisedDetector}
7+
const DeterministicDetector = Union{DeterministicSupervisedDetector,
8+
DeterministicUnsupervisedDetector}
99

1010
const StatTraits = StatisticalTraits
1111

@@ -67,8 +67,8 @@ StatTraits.fit_data_scitype(M::Type{<:SupervisedAnnotator}) =
6767
# constructor will accept it as a valid argument, which then enables
6868
# *evaluation* of the detector with labeled data:
6969
StatTraits.fit_data_scitype(M::Type{<:Union{
70-
AbstractProbabilisticUnsupervisedDetector,
71-
AbstractDeterministicUnsupervisedDetector}}) =
70+
ProbabilisticUnsupervisedDetector,
71+
DeterministicUnsupervisedDetector}}) =
7272
Union{Tuple{input_scitype(M)},
7373
Tuple{input_scitype(M),target_scitype(M)}}
7474

0 commit comments

Comments
 (0)