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

BERTCGAForcaster Class #274

Open
wants to merge 35 commits into
base: master
Choose a base branch
from
Open

BERTCGAForcaster Class #274

wants to merge 35 commits into from

Conversation

sonqt
Copy link

@sonqt sonqt commented Feb 28, 2025

Description

This PR introduces several enhancements and new features, including:

  • Forecaster Class: Implements support for auto-encoding transformer models such as BERT, RoBERTa, SpanBERT, DeBERTa, and ELECTRA.
  • New Metric for forecaster.summarize(): Adds an additional evaluation metric to improve summarization capabilities.
  • BERTCGAModel Class: Implements a new model class specifically designed for forecasting with BERT-based architectures.
  • Jupyter Notebooks:
    • Demonstration notebooks showcasing how to use BERTCGAModel.
    • Notebooks for reproducing the performance results reported in the paper.
  • Enhanced Download Configuration: Modifies download_config to enable downloading fine-tuned BERT-base and RoBERTa-large models from the Zissou server.

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.

2 participants