-
Notifications
You must be signed in to change notification settings - Fork 9
Open
Description
I can't use the diagdirichlet calibrator because there seems no way to set the the class' l2 parameter. Did it get lost in refactoring or am I using it wrong?
[3](vscode-notebook-cell:?execution_count=11&line=3) def calibrate_dirichlet(conf_calib, conf_test, labels_calib):
----> [9](vscode-notebook-cell:?execution_count=11&line=9) calibrator = DiagonalDirichletCalibrator(l2=1e-3)
[10](vscode-notebook-cell:?execution_count=11&line=10) calibrator.fit(conf_calib, labels_calib)
[12](vscode-notebook-cell:?execution_count=11&line=12) calibrated_conf = calibrator.predict(conf_test)
TypeError: FullDirichletCalibrator.__init__() got an unexpected keyword argument 'l2'
Metadata
Metadata
Assignees
Labels
No labels