Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add size-time warning on cpu #247

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

anuragg1209
Copy link
Contributor

This pull request introduces a new time usage tracking feature to the TabPFN library, which includes changes to multiple files to integrate this functionality. The main changes involve adding the TimeUsageTracker class, updating the fit and predict methods of classifiers and regressors to use this tracker, and adding warnings for CPU usage addressing issue #192

Time usage tracking:

  • src/tabpfn/model/time_tracker.py: Added the TimeUsageTracker class to track time usage for code segments, with warnings for segments running longer than a specified threshold.

Integration in TabPFNClassifier:

Integration in TabPFNRegressor:

Constants and utilities:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant