Commit d5910b31 authored by VictorSanh's avatar VictorSanh Committed by Lysandre Debut
Browse files

move xnli processor (and utils) to transformers/data/processors

parent 289cf4d2
...@@ -44,9 +44,9 @@ from transformers import (WEIGHTS_NAME, ...@@ -44,9 +44,9 @@ from transformers import (WEIGHTS_NAME,
from transformers import AdamW, WarmupLinearSchedule from transformers import AdamW, WarmupLinearSchedule
from utils_xnli import xnli_compute_metrics as compute_metrics from transformers import xnli_compute_metrics as compute_metrics
from utils_xnli import xnli_output_modes as output_modes from transformers import xnli_output_modes as output_modes
from utils_xnli import xnli_processors as processors from transformers import xnli_processors as processors
from transformers import glue_convert_examples_to_features as convert_examples_to_features from transformers import glue_convert_examples_to_features as convert_examples_to_features
......
...@@ -25,7 +25,8 @@ from .file_utils import (TRANSFORMERS_CACHE, PYTORCH_TRANSFORMERS_CACHE, PYTORCH ...@@ -25,7 +25,8 @@ from .file_utils import (TRANSFORMERS_CACHE, PYTORCH_TRANSFORMERS_CACHE, PYTORCH
from .data import (is_sklearn_available, from .data import (is_sklearn_available,
InputExample, InputFeatures, DataProcessor, InputExample, InputFeatures, DataProcessor,
glue_output_modes, glue_convert_examples_to_features, glue_output_modes, glue_convert_examples_to_features,
glue_processors, glue_tasks_num_labels) glue_processors, glue_tasks_num_labels,
xnli_output_modes, xnli_processors, xnli_tasks_num_labels)
if is_sklearn_available(): if is_sklearn_available():
from .data import glue_compute_metrics from .data import glue_compute_metrics
......
from .processors import InputExample, InputFeatures, DataProcessor from .processors import InputExample, InputFeatures, DataProcessor
from .processors import glue_output_modes, glue_processors, glue_tasks_num_labels, glue_convert_examples_to_features from .processors import glue_output_modes, glue_processors, glue_tasks_num_labels, glue_convert_examples_to_features
from .processors import xnli_output_modes, xnli_processors, xnli_tasks_num_labels
from .metrics import is_sklearn_available from .metrics import is_sklearn_available
if is_sklearn_available(): if is_sklearn_available():
......
from .utils import InputExample, InputFeatures, DataProcessor from .utils import InputExample, InputFeatures, DataProcessor
from .glue import glue_output_modes, glue_processors, glue_tasks_num_labels, glue_convert_examples_to_features from .glue import glue_output_modes, glue_processors, glue_tasks_num_labels, glue_convert_examples_to_features
from .xnli import xnli_output_modes, xnli_processors, xnli_tasks_num_labels
...@@ -20,7 +20,7 @@ from __future__ import absolute_import, division, print_function ...@@ -20,7 +20,7 @@ from __future__ import absolute_import, division, print_function
import logging import logging
import os import os
from transformers.data.processors import DataProcessor, InputExample from .utils import DataProcessor, InputExample
from transformers.data.metrics import simple_accuracy from transformers.data.metrics import simple_accuracy
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment