Skip to content

Commit

Permalink
Merge pull request #54 from dscolby/auto-juliaformatter-pr
Browse files Browse the repository at this point in the history
Automatic JuliaFormatter.jl run
  • Loading branch information
dscolby authored Jun 16, 2024
2 parents dfa8d88 + d4064b1 commit 34d13ca
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 4 deletions.
1 change: 0 additions & 1 deletion src/estimators.jl
Original file line number Diff line number Diff line change
Expand Up @@ -349,7 +349,6 @@ mutable struct DoubleMachineLearning <: CausalEstimator
iterations=round(size(X, 1) / 10),
approximator_neurons=round(size(X, 1) / 10),
)

task = var_type(Y) isa Binary ? "classification" : "regression"

return new(
Expand Down
4 changes: 1 addition & 3 deletions src/metalearners.jl
Original file line number Diff line number Diff line change
Expand Up @@ -443,9 +443,8 @@ mutable struct RLearner <: Metalearner
iterations=round(size(X, 1) / 10),
approximator_neurons=round(size(X, 1) / 10),
)

task = var_type(Y) isa Binary ? "classification" : "regression"

return new(
Float64.(X),
Float64.(T),
Expand Down Expand Up @@ -569,7 +568,6 @@ mutable struct DoublyRobustLearner <: Metalearner
iterations=round(size(X, 1) / 10),
approximator_neurons=round(size(X, 1) / 10),
)

task = var_type(Y) isa Binary ? "classification" : "regression"

return new(
Expand Down

0 comments on commit 34d13ca

Please sign in to comment.