Skip to content

Commit

Permalink
fix unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
memoryz committed Oct 28, 2023
1 parent 813a68d commit df296a6
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ abstract class BaseDiffInDiffEstimator(override val uid: String)
.setLabelCol(getOutcomeCol)
.setFitIntercept(fitIntercept)
.setLoss("squaredError")
.setRegParam(0.0)
.setRegParam(1E-10)

assembler.transform _ andThen regression.fit apply df
}
Expand Down

0 comments on commit df296a6

Please sign in to comment.