Skip to content

Implement centralized PCBM layer #21

@dgcnz

Description

@dgcnz

Currently, the PosthocLinearCBM is trained in a very weird way: Using sklearn.SGDClassifier and then copying the weights into the torch weight matrices. Why? The code says "for pedagogical reasons" but why not just using torch optim.SGD and avoid 20+ lines of unnecessary code?

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions