-
Notifications
You must be signed in to change notification settings - Fork 4
/
concrete-types.jl
203 lines (171 loc) · 5.36 KB
/
concrete-types.jl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
##### Beginning of file
struct FeatureContrastsNotYetGenerated <: AbstractNonExistentFeatureContrasts
end
struct FitNotYetRunUnderlyingObject <: AbstractNonExistentUnderlyingObject
end
struct FitFailedUnderlyingObject <: AbstractNonExistentUnderlyingObject
end
"""
"""
mutable struct GLMModel <: AbstractEstimator
name::T1 where T1 <: AbstractString
isclassificationmodel::T2 where T2 <: Bool
isregressionmodel::T3 where T3 <: Bool
formula::T4 where T4 <: Any
family::T5 where T5 <: Any
link::T6 where T6 <: Any
# parameters (learned from data):
underlyingglm::T7 where T7 <: Any
end
"""
"""
mutable struct KnetModel <: AbstractEstimator
name::T1 where T1 <: AbstractString
isclassificationmodel::T2 where T2 <: Bool
isregressionmodel::T3 where T3 <: Bool
# hyperparameters (not learned from data):
predict_function_source::T4 where T4 <: AbstractString
loss_function_source::T5 where T5 <: AbstractString
predict_function::T6 where T6 <: Any
loss_function::T7 where T7 <: Any
losshyperparameters::T8 where T8 <: AbstractDict
optimizationalgorithm::T9 where T9 <: Symbol
optimizerhyperparameters::T10 where T10 <: AbstractDict
minibatchsize::T11 where T11 <: Integer
maxepochs::T12 where T12 <: Integer
printlosseverynepochs::T13 where T13 <: Integer
# parameters (learned from data):
modelweights::T14 where T14 <: AbstractArray
modelweightoptimizers::T15 where T15 <: Any
# learning state
history::T16 where T16 <: Any
end
"""
"""
struct SimplePipeline <: AbstractPipeline
name::T1 where T1 <: AbstractString
objectsvector::T2 where T2 <: AbstractVector{Fittable}
end
struct PGFPlotsXPlot{T} <: AbstractPlot{T}
underlying_object::T
end
"""
"""
struct ImmutablePackageMultiLabelPredictionTransformer <: AbstractEstimator
label_names::T1 where T1 <: AbstractVector{Symbol}
end
"""
"""
struct ImmutablePackageSingleLabelPredictionTransformer <: AbstractEstimator
single_label_name::T1 where T1 <: Symbol
end
"""
"""
struct ImmutablePackageSingleLabelPredictProbaTransformer <:
AbstractEstimator
single_label_name::T1 where T1 <: Symbol
end
"""
"""
struct ImmutablePredictionsSingleLabelInt2StringTransformer <:
AbstractEstimator
index::T1 where T1 <: Integer
levels::T2 where T2 <: AbstractVector
end
"""
"""
struct ImmutablePredictProbaSingleLabelInt2StringTransformer <:
AbstractEstimator
index::T1 where T1 <: Integer
levels::T2 where T2 <: AbstractVector
end
"""
"""
struct DataFrameFeatureContrasts <: AbstractFeatureContrasts
columns::T1 where T1 <: AbstractVector{Symbol}
num_df_columns::T2 where T2 <: Integer
contrasts::T3 where T3 <: AbstractDict
num_array_columns::T4 where T4 <: Integer
end
"""
"""
mutable struct MutableDataFrame2DecisionTreeTransformer <: AbstractEstimator
feature_names::T1 where T1 <: AbstractVector
single_label_name::T2 where T2 <: Symbol
levels::T3 where T3 <: AbstractVector
dffeaturecontrasts::T4 where T4 <: AbstractFeatureContrasts
end
"""
"""
struct ImmutableDataFrame2GLMSingleLabelBinaryClassTransformer <:
AbstractEstimator
label::T1 where T1 <: Symbol
positive_class::T2 where T2 <: AbstractString
end
"""
"""
mutable struct MutableDataFrame2ClassificationKnetTransformer <:
AbstractEstimator
feature_names::T1 where T1 <: AbstractVector
label_names::T2 where T2 <: AbstractVector{Symbol}
label_levels::T3 where T3 <: AbstractDict
index::T4 where T4 <: Integer
transposefeatures::T5 where T5 <: Bool
transposelabels::T6 where T6 <: Bool
dffeaturecontrasts::T7 where T7 <: AbstractFeatureContrasts
end
"""
"""
mutable struct MutableDataFrame2RegressionKnetTransformer <:
AbstractEstimator
feature_names::T1 where T1 <: AbstractVector
label_names::T2 where T2 <: AbstractVector{Symbol}
transposefeatures::T3 where T3 <: Bool
transposelabels::T4 where T4 <: Bool
dffeaturecontrasts::T5 where T5 <: AbstractFeatureContrasts
function MutableDataFrame2RegressionKnetTransformer(
feature_names::AbstractVector,
label_names::AbstractVector{Symbol};
transposefeatures::Bool = true,
transposelabels::Bool = false,
)
result = new(
feature_names,
label_names,
transposefeatures,
transposelabels,
)
return result
end
end
"""
"""
struct ImmutableFeatureArrayTransposerTransformer <: AbstractEstimator
end
"""
"""
mutable struct LIBSVMModel <: AbstractEstimator
name::T1 where T1 <: AbstractString
isclassificationmodel::T2 where T2 <: Bool
isregressionmodel::T3 where T3 <: Bool
levels::T4 where T4 <: AbstractVector
# hyperparameters (not learned from data):
hyperparameters::T5 where T5 <: AbstractDict
# parameters (learned from data):
underlyingsvm::T6 where T6 <: Any
end
"""
"""
mutable struct DecisionTreeModel <:
AbstractEstimator
name::T1 where T1 <: AbstractString
isclassificationmodel::T2 where T2 <: Bool
isregressionmodel::T3 where T3 <: Bool
single_label_name::T4 where T4 <: Symbol
levels::T5 where T5 <: AbstractVector
# hyperparameters (not learned from data):
hyperparameters::T6 where T6 <: AbstractDict
# parameters (learned from data):
underlyingrandomforest::T7 where T7 <: Any
end
##### End of file