mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-03 11:41:51 +06:00
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
# coding=utf-8
|
|
# Copyright 2018 The Google AI Language Team Authors and The HuggingFace Inc. team.
|
|
# Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from .glue import (ColaProcessor,
|
|
MnliProcessor,
|
|
MnliMismatchedProcessor,
|
|
MrpcProcessor,
|
|
Sst2Processor,
|
|
StsbProcessor,
|
|
QqpProcessor,
|
|
QnliProcessor,
|
|
RteProcessor,
|
|
WnliProcessor,
|
|
convert_examples_to_glue_features,
|
|
)
|
|
|
|
from .utils import DataProcessor, simple_accuracy, acc_and_f1, pearson_and_spearman, compute_metrics
|
|
|
|
processors = {
|
|
"cola": ColaProcessor,
|
|
"mnli": MnliProcessor,
|
|
"mnli-mm": MnliMismatchedProcessor,
|
|
"mrpc": MrpcProcessor,
|
|
"sst-2": Sst2Processor,
|
|
"sts-b": StsbProcessor,
|
|
"qqp": QqpProcessor,
|
|
"qnli": QnliProcessor,
|
|
"rte": RteProcessor,
|
|
"wnli": WnliProcessor,
|
|
}
|
|
|
|
output_modes = {
|
|
"cola": "classification",
|
|
"mnli": "classification",
|
|
"mnli-mm": "classification",
|
|
"mrpc": "classification",
|
|
"sst-2": "classification",
|
|
"sts-b": "regression",
|
|
"qqp": "classification",
|
|
"qnli": "classification",
|
|
"rte": "classification",
|
|
"wnli": "classification",
|
|
}
|