PIPELINE_DATA = [ "text-classification": { name: "Text Classification", subtasks: [ { type: "acceptability-classification", name: "Acceptability Classification", }, { type: "entity-linking-classification", name: "Entity Linking Classification", }, { type: "fact-checking", name: "Fact Checking", }, { type: "intent-classification", name: "Intent Classification", }, { type: "language-identification", name: "Language Identification", }, { type: "multi-class-classification", name: "Multi Class Classification", }, { type: "multi-label-classification", name: "Multi Label Classification", }, { type: "multi-input-text-classification", name: "Multi-input Text Classification", }, { type: "natural-language-inference", name: "Natural Language Inference", }, { type: "semantic-similarity-classification", name: "Semantic Similarity Classification", }, { type: "sentiment-classification", name: "Sentiment Classification", }, { type: "topic-classification", name: "Topic Classification", }, { type: "semantic-similarity-scoring", name: "Semantic Similarity Scoring", }, { type: "sentiment-scoring", name: "Sentiment Scoring", }, { type: "sentiment-analysis", name: "Sentiment Analysis", }, { type: "hate-speech-detection", name: "Hate Speech Detection", }, { type: "text-scoring", name: "Text Scoring", }, ], modality: "nlp", color: "orange", }, "token-classification": { name: "Token Classification", subtasks: [ { type: "named-entity-recognition", name: "Named Entity Recognition", }, { type: "part-of-speech", name: "Part of Speech", }, { type: "parsing", name: "Parsing", }, { type: "lemmatization", name: "Lemmatization", }, { type: "word-sense-disambiguation", name: "Word Sense Disambiguation", }, { type: "coreference-resolution", name: "Coreference-resolution", }, ], modality: "nlp", color: "blue", }, "table-question-answering": { name: "Table Question Answering", modality: "nlp", color: "green", }, "question-answering": { name: "Question Answering", subtasks: [ { type: "extractive-qa", name: "Extractive QA", }, { type: "open-domain-qa", name: "Open Domain QA", }, { type: "closed-domain-qa", name: "Closed Domain QA", }, ], modality: "nlp", color: "blue", }, "zero-shot-classification": { name: "Zero-Shot Classification", modality: "nlp", color: "yellow", }, "translation": { name: "Translation", modality: "nlp", color: "green", }, "summarization": { name: "Summarization", subtasks: [ { type: "news-articles-summarization", name: "News Articles Summarization", }, { type: "news-articles-headline-generation", name: "News Articles Headline Generation", }, ], modality: "nlp", color: "indigo", }, "conversational": { name: "Conversational", subtasks: [ { type: "dialogue-generation", name: "Dialogue Generation", }, ], modality: "nlp", color: "green", }, "feature-extraction": { name: "Feature Extraction", modality: "multimodal", color: "red", }, "text-generation": { name: "Text Generation", subtasks: [ { type: "dialogue-modeling", name: "Dialogue Modeling", }, { type: "language-modeling", name: "Language Modeling", }, ], modality: "nlp", color: "indigo", }, "text2text-generation": { name: "Text2Text Generation", subtasks: [ { type: "text-simplification", name: "Text simplification", }, { type: "explanation-generation", name: "Explanation Generation", }, { type: "abstractive-qa", name: "Abstractive QA", }, { type: "open-domain-abstractive-qa", name: "Open Domain Abstractive QA", }, { type: "closed-domain-qa", name: "Closed Domain QA", }, { type: "open-book-qa", name: "Open Book QA", }, { type: "closed-book-qa", name: "Closed Book QA", }, ], modality: "nlp", color: "indigo", }, "fill-mask": { name: "Fill-Mask", subtasks: [ { type: "slot-filling", name: "Slot Filling", }, { type: "masked-language-modeling", name: "Masked Language Modeling", }, ], modality: "nlp", color: "red", }, "sentence-similarity": { name: "Sentence Similarity", modality: "nlp", color: "yellow", }, "text-to-speech": { name: "Text-to-Speech", modality: "audio", color: "yellow", }, "text-to-audio": { name: "Text-to-Audio", modality: "audio", color: "yellow", }, "automatic-speech-recognition": { name: "Automatic Speech Recognition", modality: "audio", color: "yellow", }, "audio-to-audio": { name: "Audio-to-Audio", modality: "audio", color: "blue", }, "audio-classification": { name: "Audio Classification", subtasks: [ { type: "keyword-spotting", name: "Keyword Spotting", }, { type: "speaker-identification", name: "Speaker Identification", }, { type: "audio-intent-classification", name: "Audio Intent Classification", }, { type: "audio-emotion-recognition", name: "Audio Emotion Recognition", }, { type: "audio-language-identification", name: "Audio Language Identification", }, ], modality: "audio", color: "green", }, "voice-activity-detection": { name: "Voice Activity Detection", modality: "audio", color: "red", }, "depth-estimation": { name: "Depth Estimation", modality: "cv", color: "yellow", }, "image-classification": { name: "Image Classification", subtasks: [ { type: "multi-label-image-classification", name: "Multi Label Image Classification", }, { type: "multi-class-image-classification", name: "Multi Class Image Classification", }, ], modality: "cv", color: "blue", }, "object-detection": { name: "Object Detection", subtasks: [ { type: "face-detection", name: "Face Detection", }, { type: "vehicle-detection", name: "Vehicle Detection", }, ], modality: "cv", color: "yellow", }, "image-segmentation": { name: "Image Segmentation", subtasks: [ { type: "instance-segmentation", name: "Instance Segmentation", }, { type: "semantic-segmentation", name: "Semantic Segmentation", }, { type: "panoptic-segmentation", name: "Panoptic Segmentation", }, ], modality: "cv", color: "green", }, "text-to-image": { name: "Text-to-Image", modality: "multimodal", color: "yellow", }, "image-to-text": { name: "Image-to-Text", subtasks: [ { type: "image-captioning", name: "Image Captioning", }, ], modality: "multimodal", color: "red", }, "image-to-image": { name: "Image-to-Image", modality: "cv", color: "indigo", }, "unconditional-image-generation": { name: "Unconditional Image Generation", modality: "cv", color: "green", }, "video-classification": { name: "Video Classification", modality: "cv", color: "blue", }, "reinforcement-learning": { name: "Reinforcement Learning", modality: "rl", color: "red", }, "robotics": { name: "Robotics", modality: "rl", subtasks: [ { type: "grasping", name: "Grasping", }, { type: "task-planning", name: "Task Planning", }, ], color: "blue", }, "tabular-classification": { name: "Tabular Classification", modality: "tabular", subtasks: [ { type: "tabular-multi-class-classification", name: "Tabular Multi Class Classification", }, { type: "tabular-multi-label-classification", name: "Tabular Multi Label Classification", }, ], color: "blue", }, "tabular-regression": { name: "Tabular Regression", modality: "tabular", subtasks: [ { type: "tabular-single-column-regression", name: "Tabular Single Column Regression", }, ], color: "blue", }, "tabular-to-text": { name: "Tabular to Text", modality: "tabular", subtasks: [ { type: "rdf-to-text", name: "RDF to text", }, ], color: "blue", hideInModels: true, }, "table-to-text": { name: "Table to Text", modality: "nlp", color: "blue", hideInModels: true, }, "multiple-choice": { name: "Multiple Choice", subtasks: [ { type: "multiple-choice-qa", name: "Multiple Choice QA", }, { type: "multiple-choice-coreference-resolution", name: "Multiple Choice Coreference Resolution", }, ], modality: "nlp", color: "blue", hideInModels: true, }, "text-retrieval": { name: "Text Retrieval", subtasks: [ { type: "document-retrieval", name: "Document Retrieval", }, { type: "utterance-retrieval", name: "Utterance Retrieval", }, { type: "entity-linking-retrieval", name: "Entity Linking Retrieval", }, { type: "fact-checking-retrieval", name: "Fact Checking Retrieval", }, ], modality: "nlp", color: "indigo", hideInModels: true, }, "time-series-forecasting": { name: "Time Series Forecasting", modality: "tabular", subtasks: [ { type: "univariate-time-series-forecasting", name: "Univariate Time Series Forecasting", }, { type: "multivariate-time-series-forecasting", name: "Multivariate Time Series Forecasting", }, ], color: "blue", hideInModels: true, }, "text-to-video": { name: "Text-to-Video", modality: "multimodal", color: "green", }, "visual-question-answering": { name: "Visual Question Answering", subtasks: [ { type: "visual-question-answering", name: "Visual Question Answering", }, ], modality: "multimodal", color: "red", }, "document-question-answering": { name: "Document Question Answering", subtasks: [ { type: "document-question-answering", name: "Document Question Answering", }, ], modality: "multimodal", color: "blue", hideInDatasets: true, }, "zero-shot-image-classification": { name: "Zero-Shot Image Classification", modality: "cv", color: "yellow", }, "graph-ml": { name: "Graph Machine Learning", modality: "multimodal", color: "green", }, "other": { name: "Other", modality: "other", color: "blue", hideInModels: true, hideInDatasets: true, }, ]