Skip to content

Commit 8c0b05a

Browse files
committed
✍🏻 Rename report variable to better
1 parent c17b5a5 commit 8c0b05a

File tree

3 files changed

+18
-18
lines changed

3 files changed

+18
-18
lines changed

src/encoders/missingness_encoding/interface_mlj.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ end;
2828

2929
# 4. Fitted parameters (for user access)
3030
MMI.fitted_params(::MissingnessEncoder, fitresult) = (
31-
new_cat_given_col_val = fitresult,
31+
label_for_missing_given_feature = fitresult,
3232
)
3333

3434
# 5. Fit method
@@ -40,7 +40,7 @@ function MMI.fit(transformer::MissingnessEncoder, verbosity::Int, X)
4040
ordered_factor = transformer.ordered_factor,
4141
label_for_missing = transformer.label_for_missing,
4242
)
43-
fitresult = generic_cache[:new_cat_given_col_val]
43+
fitresult = generic_cache[:label_for_missing_given_feature]
4444

4545
report = (encoded_features = generic_cache[:encoded_features],) # report only has list of encoded features
4646
cache = nothing
@@ -51,7 +51,7 @@ end;
5151
# 6. Transform method
5252
function MMI.transform(transformer::MissingnessEncoder, fitresult, Xnew)
5353
generic_cache = Dict(
54-
:new_cat_given_col_val =>
54+
:label_for_missing_given_feature =>
5555
fitresult,
5656
)
5757
Xnew_transf = missingness_encoder_transform(Xnew, generic_cache)
@@ -119,7 +119,7 @@ and if the raw type subtypes `Number` then the new value is the lowest value in
119119
120120
The fields of `fitted_params(mach)` are:
121121
122-
- `new_cat_given_col_val`: A dictionary that for each column, maps `missing` into some value according to `label_for_missing`
122+
- `label_for_missing_given_feature`: A dictionary that for each column, maps `missing` into some value according to `label_for_missing`
123123
124124
# Report
125125

src/encoders/missingness_encoding/missingness_encoding.jl

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ types that are in `Char`, `AbstractString`, and `Number`.
2222
2323
# Returns (in a dict)
2424
25-
- `new_cat_given_col_val`: A dictionary that for each column, maps `missing` into some value according to `label_for_missing`
25+
- `label_for_missing_given_feature`: A dictionary that for each column, maps `missing` into some value according to `label_for_missing`
2626
- `encoded_features`: The subset of the categorical features of X that were encoded
2727
"""
2828
function missingness_encoder_fit(
@@ -74,26 +74,26 @@ function missingness_encoder_fit(
7474
end
7575

7676
# Nonmissing levels remain as is
77-
new_cat_given_col_val = Dict{Missing, col_type}()
77+
label_for_missing_given_feature = Dict{Missing, col_type}()
7878

7979
# Missing levels are mapped
8080
if elgrandtype in keys(label_for_missing)
81-
new_cat_given_col_val[missing] = label_for_missing[elgrandtype]
81+
label_for_missing_given_feature[missing] = label_for_missing[elgrandtype]
8282
elseif elgrandtype == Number
83-
new_cat_given_col_val[missing] = minimum(feat_levels) - 1
83+
label_for_missing_given_feature[missing] = minimum(feat_levels) - 1
8484
else
8585
throw(ArgumentError(UNSPECIFIED_COL_TYPE_ME(col_type, label_for_missing)))
8686
end
8787

88-
return new_cat_given_col_val::Dict{Missing, col_type}
88+
return label_for_missing_given_feature::Dict{Missing, col_type}
8989
end
9090

9191
# 2. Pass it to generic_fit
92-
new_cat_given_col_val, encoded_features = generic_fit(
92+
label_for_missing_given_feature, encoded_features = generic_fit(
9393
X, features; ignore = ignore, ordered_factor = ordered_factor, feature_mapper = feature_mapper,
9494
)
9595
cache = Dict(
96-
:new_cat_given_col_val => new_cat_given_col_val,
96+
:label_for_missing_given_feature => label_for_missing_given_feature,
9797
:encoded_features => encoded_features,
9898
)
9999
return cache
@@ -115,7 +115,7 @@ Apply a fitted missingness encoder to a table given the output of `missingness_e
115115
- `X_tr`: The table with selected features after the selected features are transformed by missingness encoder
116116
"""
117117
function missingness_encoder_transform(X, cache::Dict)
118-
new_cat_given_col_val = cache[:new_cat_given_col_val]
119-
return generic_transform(X, new_cat_given_col_val; ignore_unknown = true)
118+
label_for_missing_given_feature = cache[:label_for_missing_given_feature]
119+
return generic_transform(X, label_for_missing_given_feature; ignore_unknown = true)
120120
end
121121

test/encoders/missingness_encoding.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ end
2828
@testset "Default for Numbers Set Correctly" begin
2929
X = generate_X_with_missingness()
3030
cache = missingness_encoder_fit(X)
31-
new_cat_given_col_val = cache[:new_cat_given_col_val]
32-
@test new_cat_given_col_val[:C][missing] == minimum(levels(X.C)) - 1
31+
label_for_missing_given_feature = cache[:label_for_missing_given_feature]
32+
@test label_for_missing_given_feature[:C][missing] == minimum(levels(X.C)) - 1
3333
end
3434

3535

@@ -57,7 +57,7 @@ end
5757
result = missingness_encoder_fit(
5858
X;
5959
label_for_missing = Dict(AbstractString => "MissingOne", Char => 'X', Number => -90),
60-
)[:new_cat_given_col_val]
60+
)[:label_for_missing_given_feature]
6161

6262
true_output = Dict{Symbol, Dict{Any, Any}}(
6363
:A => Dict([(missing, "MissingOne")]),
@@ -148,8 +148,8 @@ end
148148
@test isequal(X_transf, Xnew_transf)
149149

150150
# fitted parameters is correct
151-
new_cat_given_col_val = fitted_params(mach).new_cat_given_col_val
152-
@test new_cat_given_col_val == generic_cache[:new_cat_given_col_val]
151+
label_for_missing_given_feature = fitted_params(mach).label_for_missing_given_feature
152+
@test label_for_missing_given_feature == generic_cache[:label_for_missing_given_feature]
153153

154154
# Test report
155155
@test report(mach) == (encoded_features = generic_cache[:encoded_features],)

0 commit comments

Comments
 (0)