Skip to content
This repository was archived by the owner on Apr 24, 2024. It is now read-only.
This repository was archived by the owner on Apr 24, 2024. It is now read-only.

torch.Tensor support for methods in numpy submodule #63

Description

@agoscinski

In #62 we introduce tox testing environments for metatensor-core with numpy arrays and with torch tenors. For now in tests/equisolve_tests/numpy/utilities.py we set use_torch to False, so all tests are not run using numpy as array backend. But we probably also want to support torch tensors for modules like Ridge and SparseKRR. Internally everything will be converted to numpy arrays (which we eventually change also at some point).

This would go hand in hand with a renaming of the numpy submodule to something less specific. One could use the term shallow for shallow ML methods like linear and kernel models. Or we split it up into kernel_ridge and linear_model as scikit-learn does it.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions