Skip to content

Commit

Permalink
Merge pull request #199 from IINemo/monotonic_calibration
Browse files Browse the repository at this point in the history
Monotonic calibration
  • Loading branch information
rvashurin authored Jul 3, 2024
2 parents 6fda7e3 + 4b58a9c commit 12ce596
Show file tree
Hide file tree
Showing 15 changed files with 1,249 additions and 272 deletions.
1 change: 0 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
src/lm_polygraph/app/node_modules
**/.DStore
workdir/
**/normalization/*

*.npy
*.csv
Expand Down
58 changes: 58 additions & 0 deletions examples/configs/normalization/fit/default.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
hydra:
run:
dir: ${cache_path}/${now:%Y-%m-%d}/${now:%H-%M-%S}

cache_path: ./workdir/output/normalization
save_path: '${hydra:run.dir}'

normalization_methods:
- min_max
- quantile
- binned_pcc
- isotonic_pcc

num_bins: 100

man_paths:
- polygraph_tacl_stablelm12b_coqa_train.man
- polygraph_tacl_stablelm12b_gsm8k_train.man
- polygraph_tacl_stablelm12b_mmlu_train.man
- polygraph_tacl_stablelm12b_triviaqa_train.man
- polygraph_tacl_stablelm12b_wmt14_train.man
- polygraph_tacl_stablelm12b_wmt19_train.man
- polygraph_tacl_stablelm12b_xsum_train.man

ue_method_names:
- MaximumSequenceProbability
- Perplexity
- MeanTokenEntropy
- MeanPointwiseMutualInformation
- MeanConditionalPointwiseMutualInformation
- CCP
- PTrue
- PTrueSampling
- MonteCarloSequenceEntropy
- MonteCarloNormalizedSequenceEntropy
- LexicalSimilarity_rouge1
- LexicalSimilarity_rouge2
- LexicalSimilarity_rougeL
- LexicalSimilarity_BLEU
- NumSemSets
- EigValLaplacian_NLI_score_entail
- EigValLaplacian_NLI_score_contra
- EigValLaplacian_Jaccard_score
- DegMat_NLI_score_entail
- DegMat_NLI_score_contra
- DegMat_Jaccard_score
- Eccentricity_NLI_score_entail
- Eccentricity_NLI_score_contra
- Eccentricity_Jaccard_score
- SemanticEntropy
- SAR
- TokenSAR
- SentenceSAR
- RenyiNeg
- FisherRao

gen_metric_names:
- AlignScore
Loading

0 comments on commit 12ce596

Please sign in to comment.