MCPcopy Create free account

hub / github.com/Snowflake-Labs/ReFoRCE / types & classes

Types & classes83 in github.com/Snowflake-Labs/ReFoRCE

↓ 4 callersClassSqlEnv
methods/ReFoRCE/sql.py:12
↓ 3 callersClassGPTBigCodeModel
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:598
↓ 3 callersClassSchemaItemClassifierInference
spider2-lite/baselines/codes/schema_item_filter.py:275
↓ 2 callersClassGPTBigCodeAttention
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:160
↓ 2 callersClassGPTChat
methods/ReFoRCE/chat.py:58
↓ 2 callersClassLinker
methods/SL/gen_sl/naive_parser.py:14
↓ 2 callersClassREFORCE
methods/ReFoRCE/agent.py:14
↓ 1 callersClassColumn
spider2-lite/baselines/dailsql/utils/datasets/spider.py:65
↓ 1 callersClassCoreNLP
spider2-lite/baselines/dailsql/utils/linking_utils/corenlp.py:30
↓ 1 callersClassFlashSelfAttention
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 callersClassFocalLoss
spider2-lite/baselines/codes/utils/classifier_loss.py:7
↓ 1 callersClassGPTBigCodeBlock
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:380
↓ 1 callersClassGPTBigCodeMLP
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:362
↓ 1 callersClassGloVe
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:43
↓ 1 callersClassMatch
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:57
↓ 1 callersClassPreprocessedSchema
spider2-lite/baselines/dailsql/utils/linking_process.py:18
↓ 1 callersClassPretrainDataset
spider2-lite/baselines/codes/utils/load_pt_dataset.py:49
↓ 1 callersClassPrompts
methods/ReFoRCE/prompt.py:28
↓ 1 callersClassSFTSQLGenerationDataset
spider2-lite/baselines/codes/utils/load_sft_dataset.py:100
↓ 1 callersClassSchema
spider2-lite/baselines/dailsql/utils/datasets/spider.py:86
↓ 1 callersClassSchemaItemClassifier
spider2-lite/baselines/codes/utils/classifier_model.py:6
↓ 1 callersClassSpider2CForDailSQL_Dataset
spider2-lite/baselines/dailsql/utils/data_builder.py:232
↓ 1 callersClassSpiderEncoderV2Preproc
spider2-lite/baselines/dailsql/utils/linking_process.py:108
↓ 1 callersClassSqliteTable
spider2-lite/baselines/dailsql/utils/utils.py:13
↓ 1 callersClassTable
spider2-lite/baselines/dailsql/utils/datasets/spider.py:76
↓ 1 callersClassTeeOutput
spider2-lite/evaluation_suite/evaluate.py:20
↓ 1 callersClassTeeOutput
spider2-snow/evaluation_suite/evaluate.py:23
↓ 1 callersClassTimeoutException
spider2-lite/evaluation_suite/evaluate_utils.py:22
↓ 1 callersClassTimeoutException
spider2-snow/evaluation_suite/evaluate_utils.py:22
ClassAbstractPreproc
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
ClassBaseChat
methods/ReFoRCE/chat.py:5
ClassBaselinePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:175
ClassBaselineQuestionSqlExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:64
ClassBaselineWithoutForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:296
ClassBasicDataset
spider2-lite/baselines/dailsql/utils/data_builder.py:11
ClassBasicExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:8
ClassBasicICLPrompt
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:5
ClassBasicPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:11
ClassCBRPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:549
ClassClassifierLoss
spider2-lite/baselines/codes/utils/classifier_loss.py:30
ClassCompleteExampleStyle
Examples are in the same format as target question
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:40
ClassCosineSimilarExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:45
ClassEXAMPLE_TYPE
spider2-lite/baselines/dailsql/utils/enums.py:22
ClassEmbedder
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:15
ClassEuclideanDistanceExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:82
ClassEuclideanDistancePreSkeletonSimilarPlusSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:288
ClassEuclideanDistancePreSkeletonSimilarThresholdSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:240
ClassEuclideanDistanceQuestionMaskPreSkeletonSimilarThresholdSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:321
ClassEuclideanDistanceQuestionMaskPreSkeletonSimilarThresholdShiftSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:373
ClassEuclideanDistanceQuestionMaskSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:204
ClassEuclideanDistanceSkeletonSimilarThresholdSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:152
ClassEuclideanDistanceThresholdExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:114
ClassGPTBigCodeForCausalLM
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:820
ClassGPTBigCodeForSequenceClassification
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:973
ClassGPTBigCodeForTokenClassification
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:1091
ClassGPTBigCodePreTrainedModel
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
ClassInstructionCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:521
ClassInstructionPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:206
ClassInstructionWithForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:321
ClassInstructionWithRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:420
ClassLLM
spider2-lite/baselines/dailsql/utils/enums.py:45
ClassLinearWarmupCosineAnnealingLR
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
ClassNumberSignCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:494
ClassNumberSignPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:151
ClassNumberSignQuestionSqlExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:51
ClassNumberSignWithForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:261
ClassNumberSignWithoutRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:396
ClassPromptClass
spider2-lite/baselines/dailsql/prompt/prompt_builder.py:103
ClassQuestionSqlExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:15
ClassQuestionSqlWithRuleExampleStyle
Provide QA pair as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:27
ClassREPR_TYPE
spider2-lite/baselines/dailsql/utils/enums.py:1
ClassRandomExampleSelector
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:29
ClassSELECTOR_TYPE
spider2-lite/baselines/dailsql/utils/enums.py:31
ClassSQLCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:445
ClassSQLPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:28
ClassSQLWithRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:353
ClassSchemaItemClassifierDataset
spider2-lite/baselines/codes/utils/load_classifier_dataset.py:5
ClassSpiderItem
spider2-lite/baselines/dailsql/utils/datasets/spider.py:56
ClassSqlExampleStyle
Only show sqls as examples
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:4
ClassTextCOTPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:470
ClassTextPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:130
ClassTextWithForeignKeyPrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:231
ClassTextWithRulePrompt
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:375