PyTorch Probing

Python module for creating and using probes with PyTorch modules.

This project was developed as part of the Cognitive Architectures research line from the Hub for Artificial Intelligence and Cognitive Architectures (H.IAAC) of the State University of Campinas (UNICAMP). See more projects from the group here.

Repository Structure

  • src: pytorch_probing source code

  • docs: documentation files

  • examples: examples of usage (see the documentation)

  • test: test codes

Requirements

  • Python

  • Other dependencies are automatically installed with pip

Installation / Usage

Installation

  • Installing using pip:

    pip install --upgrade pip
pip install pytorch_probing

  • For installing from the repository:

    git clone https://github.com/H-IAAC/pytorch_probing
cd pytorch_probing
pip install --upgrade pip
pip install .

Usage

See the documentation for usage and examples.

Citation

@software{pytorch_probing,
author = {Cardoso do Nascimento, Elton and Dornhofer Paro Costa, Paula},
doi = {10.5281/zenodo.13629705},
title = {PyTorch Probing},
url = {hhttps://github.com/H-IAAC/pytorch-probing}
}

Authors

  • (2024-) EltonCN : M. Eng. student, FEEC-Unicamp

  • (Advisor, 2024-) Paula Dornhofer Paro Costa: Professor, FEEC-UNICAMP

Acknowledgements

Project supported by the brazilian Ministry of Science, Technology and Innovations, with resources from Law No. 8,248, of October 23, 1991

Licence

Copyright 2024 H.IAAC

Licensed under the GNU LESSER GENERAL PUBLIC LICENSE, Version 3 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

   http://www.gnu.org/licenses/lgpl-3.0.html

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.