Skip to content

Model and ablation test updates#1

Merged
dingxuan92 merged 1 commit intodingxuan92:ebcl-model-implementationfrom
Sukiii17:ebcl-model-implementation
Apr 17, 2026
Merged

Model and ablation test updates#1
dingxuan92 merged 1 commit intodingxuan92:ebcl-model-implementationfrom
Sukiii17:ebcl-model-implementation

Conversation

@Sukiii17
Copy link
Copy Markdown

  1. Add a simple ablation test with synthetic data.
  2. Modify the model and unit test add the followings:
    a. Add more docstring and slightly modify naming and styling for documation and code quality purpose.
    b. Change the embedding dimension and feedforward dimension to the same as in paper section 4.2 Model Architecture.
    c. Add Fusion Self-Attention pooling before Linear projection to be the same as in paper section 4.2 Model Architecture.
    d. Add more more premeter supervised_uses_post to determine if prediction will use pre embedding or pre+post embedding as described in the Finetuning part of Section 3.2

2. Modify the model and unit test add the followings:
a. Add more docstring and slightly modify naming and styling for documation and code quality purpose.
b. Change the embedding dimension and feedforward dimension to the same as in paper section 4.2 Model Architecture.
c. Add Fusion Self-Attention pooling before Linear projection to be the same as in paper section 4.2 Model Architecture.
d. Add more more premeter supervised_uses_post to determine if prediction will use pre embedding or pre+post embedding as described in the Finetuning part of Section 3.2
@dingxuan92 dingxuan92 merged commit 37d7605 into dingxuan92:ebcl-model-implementation Apr 17, 2026
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