Subhalingam D 5ee9e786d1 Fix default behaviour in TextClassificationPipeline for regression problem type (#34066) ...

* update code

* update docstrings

* update tests

2024-10-15 13:06:20 +01:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_pipelines_audio_classification.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_automatic_speech_recognition.py

[TESTS] ASR pipeline (#33925)

2024-10-10 17:31:22 +02:00

test_pipelines_common.py

Pipeline: no side-effects on model.config and model.generation_config 🔫 (#33480)

2024-09-18 15:43:06 +01:00

test_pipelines_depth_estimation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_document_question_answering.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_feature_extraction.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_fill_mask.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_classification.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_feature_extraction.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_segmentation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_to_image.py

Allow FP16 or other precision inference for Pipelines (#31342)

2024-07-05 17:21:50 +01:00

test_pipelines_image_to_text.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_mask_generation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_object_detection.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_question_answering.py

Sync QuestionAnsweringPipeline (#34039)

2024-10-10 13:38:14 +01:00

test_pipelines_summarization.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_table_question_answering.py

Allow FP16 or other precision inference for Pipelines (#31342)

2024-07-05 17:21:50 +01:00

test_pipelines_text2text_generation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_text_classification.py

Fix default behaviour in TextClassificationPipeline for regression problem type (#34066)

2024-10-15 13:06:20 +01:00

test_pipelines_text_generation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_text_to_audio.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_token_classification.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_translation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_video_classification.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_visual_question_answering.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_zero_shot_audio_classification.py

Allow FP16 or other precision inference for Pipelines (#31342)

2024-07-05 17:21:50 +01:00

test_pipelines_zero_shot_image_classification.py

Image pipelines spec compliance (#33899)

2024-10-08 13:34:28 +01:00

test_pipelines_zero_shot_object_detection.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_zero_shot.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00