Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Snowflake-Labs/ReFoRCE
/ types & classes
Types & classes
83 in github.com/Snowflake-Labs/ReFoRCE
⨍
Functions
554
◇
Types & classes
83
↳
Endpoints
1
↓ 4 callers
Class
SqlEnv
methods/ReFoRCE/sql.py:12
↓ 3 callers
Class
GPTBigCodeModel
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:598
↓ 3 callers
Class
SchemaItemClassifierInference
spider2-lite/baselines/codes/schema_item_filter.py:275
↓ 2 callers
Class
GPTBigCodeAttention
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:160
↓ 2 callers
Class
GPTChat
methods/ReFoRCE/chat.py:58
↓ 2 callers
Class
Linker
methods/SL/gen_sl/naive_parser.py:14
↓ 2 callers
Class
REFORCE
methods/ReFoRCE/agent.py:14
↓ 1 callers
Class
Column
spider2-lite/baselines/dailsql/utils/datasets/spider.py:65
↓ 1 callers
Class
CoreNLP
spider2-lite/baselines/dailsql/utils/linking_utils/corenlp.py:30
↓ 1 callers
Class
FlashSelfAttention
Implement the scaled dot product attention with softmax. Arguments --------- softmax_scale: The temperature to use for the softmax att
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:99
↓ 1 callers
Class
FocalLoss
spider2-lite/baselines/codes/utils/classifier_loss.py:7
↓ 1 callers
Class
GPTBigCodeBlock
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:380
↓ 1 callers
Class
GPTBigCodeMLP
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:362
↓ 1 callers
Class
GloVe
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:43
↓ 1 callers
Class
Match
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:57
↓ 1 callers
Class
PreprocessedSchema
spider2-lite/baselines/dailsql/utils/linking_process.py:18
↓ 1 callers
Class
PretrainDataset
spider2-lite/baselines/codes/utils/load_pt_dataset.py:49
↓ 1 callers
Class
Prompts
methods/ReFoRCE/prompt.py:28
↓ 1 callers
Class
SFTSQLGenerationDataset
spider2-lite/baselines/codes/utils/load_sft_dataset.py:100
↓ 1 callers
Class
Schema
spider2-lite/baselines/dailsql/utils/datasets/spider.py:86
↓ 1 callers
Class
SchemaItemClassifier
spider2-lite/baselines/codes/utils/classifier_model.py:6
↓ 1 callers
Class
Spider2CForDailSQL_Dataset
spider2-lite/baselines/dailsql/utils/data_builder.py:232
↓ 1 callers
Class
SpiderEncoderV2Preproc
spider2-lite/baselines/dailsql/utils/linking_process.py:108
↓ 1 callers
Class
SqliteTable
spider2-lite/baselines/dailsql/utils/utils.py:13
↓ 1 callers
Class
Table
spider2-lite/baselines/dailsql/utils/datasets/spider.py:76
↓ 1 callers
Class
TeeOutput
spider2-lite/evaluation_suite/evaluate.py:20
↓ 1 callers
Class
TeeOutput
spider2-snow/evaluation_suite/evaluate.py:23
↓ 1 callers
Class
TimeoutException
spider2-lite/evaluation_suite/evaluate_utils.py:22
↓ 1 callers
Class
TimeoutException
spider2-snow/evaluation_suite/evaluate_utils.py:22
Class
AbstractPreproc
Used for preprocessing data according to the model's liking. Some tasks normally performed here: - Constructing a vocabulary from the training
spider2-lite/baselines/dailsql/utils/linking_utils/abstract_preproc.py:4
Class
BaseChat
methods/ReFoRCE/chat.py:5
Class
BaselinePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:175
Class
BaselineQuestionSqlExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:64
Class
BaselineWithoutForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:296
Class
BasicDataset
spider2-lite/baselines/dailsql/utils/data_builder.py:11
Class
BasicExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:8
Class
BasicICLPrompt
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:5
Class
BasicPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:11
Class
CBRPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:549
Class
ClassifierLoss
spider2-lite/baselines/codes/utils/classifier_loss.py:30
Class
CompleteExampleStyle
Examples are in the same format as target question
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:40
Class
CosineSimilarExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:45
Class
EXAMPLE_TYPE
spider2-lite/baselines/dailsql/utils/enums.py:22
Class
Embedder
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:15
Class
EuclideanDistanceExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:82
Class
EuclideanDistancePreSkeletonSimilarPlusSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:288
Class
EuclideanDistancePreSkeletonSimilarThresholdSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:240
Class
EuclideanDistanceQuestionMaskPreSkeletonSimilarThresholdSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:321
Class
EuclideanDistanceQuestionMaskPreSkeletonSimilarThresholdShiftSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:373
Class
EuclideanDistanceQuestionMaskSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:204
Class
EuclideanDistanceSkeletonSimilarThresholdSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:152
Class
EuclideanDistanceThresholdExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:114
Class
GPTBigCodeForCausalLM
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:820
Class
GPTBigCodeForSequenceClassification
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:973
Class
GPTBigCodeForTokenClassification
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:1091
Class
GPTBigCodePreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:462
Class
InstructionCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:521
Class
InstructionPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:206
Class
InstructionWithForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:321
Class
InstructionWithRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:420
Class
LLM
spider2-lite/baselines/dailsql/utils/enums.py:45
Class
LinearWarmupCosineAnnealingLR
Sets the learning rate of each parameter group to follow a linear warmup schedule between warmup_start_lr and base_lr followed by a cosine anneali
spider2-lite/baselines/codes/utils/lr_scheduler.py:12
Class
NumberSignCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:494
Class
NumberSignPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:151
Class
NumberSignQuestionSqlExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:51
Class
NumberSignWithForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:261
Class
NumberSignWithoutRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:396
Class
PromptClass
spider2-lite/baselines/dailsql/prompt/prompt_builder.py:103
Class
QuestionSqlExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:15
Class
QuestionSqlWithRuleExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:27
Class
REPR_TYPE
spider2-lite/baselines/dailsql/utils/enums.py:1
Class
RandomExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:29
Class
SELECTOR_TYPE
spider2-lite/baselines/dailsql/utils/enums.py:31
Class
SQLCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:445
Class
SQLPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:28
Class
SQLWithRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:353
Class
SchemaItemClassifierDataset
spider2-lite/baselines/codes/utils/load_classifier_dataset.py:5
Class
SpiderItem
spider2-lite/baselines/dailsql/utils/datasets/spider.py:56
Class
SqlExampleStyle
Only show sqls as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:4
Class
TextCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:470
Class
TextPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:130
Class
TextWithForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:231
Class
TextWithRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:375