Skip to content

self.l2 in diagdirichlet? #18

@vietcuongngoVNG

Description

@vietcuongngoVNG

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

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions