23 Repos
End-to-end training of machine learning models specifically for text data, including multilingual support.
Distinct from Model Training Pipelines: The candidates are too narrow (translation) or too general (generic model training pipelines).
Explore 23 awesome GitHub repositories matching artificial intelligence & ml · Text Model Training. Refine with filters or upvote what's useful.
fastText is a library and framework for word embedding generation, text vectorization, and supervised text classification. It provides tools to transform raw text into fixed-length vector representations and to train models that assign category labels to sentences or documents. The system utilizes subword-based vectorization and character n-gram embeddings, allowing it to generate meaningful vectors for words that were not present during training. To manage resource usage, it includes a quantized language model implementation that employs product quantization and dimensionality reduction to d
Trains supervised models to categorize text and evaluates them using precision and recall.
AutoGluon is an automated machine learning framework and multimodal library designed to automate the end-to-end pipeline from data preprocessing to high-accuracy model training and validation. It functions as an automated model trainer for tabular, image, text, and time series data, as well as a tool for time series forecasting and foundation model finetuning. The project is distinguished by its ability to jointly process and fuse different data types, allowing for the construction of multimodal neural networks that integrate images, text, and structured tables. It supports zero-shot inferenc
Trains high-quality models for text data using various backbones and supports cross-lingual transfer.
This project is a Transformer machine translation model and attention-based neural network implemented using the PyTorch deep learning framework. It functions as a text-to-text translation tool designed to convert source sequences into target language text. The implementation focuses on neural machine translation, covering the development of sequence-to-sequence architectures. It includes the full pipeline for translation, from text sequence preprocessing and vocabulary creation to model training and text generation inference. The system incorporates standard transformer components such as a
Provides the machinery for end-to-end training of text-based sequence-to-sequence models.
This project is a toolkit for fine-tuning and managing text-to-image diffusion models. It focuses on low-rank adaptation to create small, portable weight files that customize model styles and behaviors without modifying the entire base model. The project provides specialized utilities for model distillation using singular value decomposition to extract adapters from fully trained models, as well as tools for blending and merging multiple adapters through weight interpolation. It includes capabilities for subject inversion and pivotal tuning to increase the visual fidelity of specific identiti
Implements training capabilities that link images with text descriptions to improve generation accuracy.
nlp.js is a JavaScript natural language processing library and development framework used to build natural language understanding engines. It provides a toolkit for creating local machine learning models for intent classification and acts as a multilingual text processor that detects languages and normalizes text across various dialects. The framework distinguishes itself by supporting local execution on both servers and mobile devices, enabling chatbot functionality without an internet connection. It features a specialized system for conversational slot filling to collect mandatory informati
Enables the initialization of model knowledge bases by fetching JSON corpora from remote URLs.
Dies ist ein Machine-Learning-Framework zur Behandlung diverser Natural-Language-Processing-Aufgaben als einheitliches Text-zu-Text-Problem. Es bietet ein Toolkit für das Pre-Training und Fine-Tuning groß angelegter Transformer-Modelle und nutzt ein System, bei dem sowohl Eingaben als auch Ausgaben als rohe Textsequenzen formatiert sind. Das Framework zeichnet sich durch sein verteiltes Trainingssystem aus, das Mesh-basierte Strategien verwendet, um Modellgewichte und Trainings-Batches über mehrere TPU-Kerne hinweg zu skalieren. Es unterstützt Multi-Task-Learning durch die Kombination diverser Datensätze in einem einzigen Trainingsstrom unter Verwendung konfigurierbarer Mischraten, wodurch ein einziges Modell verschiedene Sprachaufgaben bewältigen kann. Das System deckt ein breites Spektrum an Funktionen ab, einschließlich Encoder-Decoder-Architekturen, Beam-Search-Decoding für Textgenerierung und Transfer-Learning-Workflows. Es enthält Dienstprogramme für die Vorbereitung von NLP-Datensätzen, die Bewertung der Modellleistung und den Export trainierter Checkpoints für die produktive Bereitstellung. Die Bibliothek unterstützt das Laden vortrainierter Modell-Checkpoints verschiedener Größen, um die Entwicklung zu beschleunigen.
Provides a unified framework for training transformer models to handle diverse text-to-text natural language processing tasks.
pycorrector is an open-source toolkit for detecting and correcting spelling and grammar errors in Chinese text. It combines multiple correction approaches, including rule-based methods using Kenlm n-gram language models and confusion sets, as well as deep learning correctors built on BERT, GPT, and T5 models. The toolkit also provides a command-line interface for batch processing Chinese text files with configurable detection and output options. The project distinguishes itself by offering a range of correction strategies that can be mixed and matched. Rule-based correction uses character-lev
Provides training scripts for BERT models on paired error-correction data to learn character-level corrections.
This project is a neural machine translation system used to build models that automatically translate text from one language to another. It utilizes sequence-to-sequence modeling to transform variable-length input sequences into corresponding output sequences. The system implements bidirectional recurrent neural network encoding and attention mechanisms to capture contextual information and focus on specific parts of the source text during translation. To manage training and inference, it employs separate computational graphs and supports distributing model layers across multiple GPU devices.
Processes raw text into batched and padded tensors using vocabulary lookups for model input.
This project is a collection of educational resources and technical guides focused on the development and implementation of large language models. It provides a comprehensive curriculum covering transformer architectures, training methods, and deployment strategies. The materials provide detailed instructions for building autonomous agents using reasoning loops and tool integration, as well as guides for fine-tuning models through supervised learning and preference optimization. It also includes tutorials for constructing retrieval augmented generation pipelines and implementing transformer m
Trains transformer-based models on sentiment datasets for text classification tasks.
PyText is an extensible PyTorch-based framework for building, training, and deploying custom natural language processing models, including text classifiers, sequence taggers, and intent-slot predictors. It provides a modular toolkit that allows developers to assemble these models using pluggable registries for model architectures, data formats, and tensorizers, all configurable through YAML files without requiring code changes. The framework distinguishes itself through its comprehensive support for the full NLP model lifecycle, from training to production inference. It includes pre-built neu
Trains deep-learning models to sort text into predefined categories using configurable architectures.
Torchtune is a PyTorch-native library for fine-tuning, aligning, and quantizing large language models. It provides a config-driven system for instantiating components, orchestrating distributed training, and managing parameter-efficient fine-tuning with quantization support, all through YAML-based configurations and command-line overrides. The library distinguishes itself through its comprehensive post-training workflow orchestration, combining supervised fine-tuning, preference optimization (DPO, PPO, GRPO), knowledge distillation, and quantization-aware training in a single configurable pip
Reads conversational data from local files or remote HTTPS URLs using the Hugging Face datasets loader.
Dieses Projekt ist eine TensorFlow-Implementierung eines Convolutional Neural Network, das für die Textklassifizierung entwickelt wurde. Es fungiert als Deep-Learning-Textkategorisierer, der Textdokumenten vordefinierte Labels zuweist, indem er gelernte Muster innerhalb von Trainingssets identifiziert und analysiert. Das Modell nutzt eine Sequenz aus Embedding-Layer-Vektorisierung, Convolutional Layers zur Merkmalsextraktion und Max-Pooling-Downsampling zur Verarbeitung von Textdaten. Die endgültigen Kategoriewahrscheinlichkeiten werden durch ein Dense-Layer-Klassifizierungssystem bestimmt. Der Workflow deckt den gesamten Machine-Learning-Lebenszyklus ab, einschließlich überwachtem Textlernen, Modelltraining mit konfigurierbaren Embedding-Dimensionen und Filtergrößen sowie Leistungsbewertung unter Verwendung von Validierungsdatensätzen und checkpoint-basierter Modellpersistenz.
Trains deep-learning models to sort text into categories using a configurable convolutional architecture.
BLIP is a vision-language model framework that combines contrastive, matching, and language modeling objectives to align images with text. Built on a multimodal encoder-decoder architecture, it supports distributed data-parallel training with cosine learning rate scheduling and sliding-window metric tracking for training stability. The framework provides capabilities for image captioning, visual question answering, and cross-modal retrieval, scoring semantic alignment between images and text through learned embeddings. It includes toolkits for fine-tuning pre-trained models on custom datasets
Trains vision-language models to generate descriptive captions for images using paired image-caption datasets.
Neuraltalk is an automated image captioning system that generates natural language descriptions for images. It utilizes a deep learning model that integrates a pretrained convolutional neural network for visual feature extraction with a recurrent neural network decoder to produce text sequences. The project provides a full workflow for training and evaluating captioning models, including weight optimization via backpropagation and gradient descent. It includes tools for measuring caption accuracy by comparing generated text against reference descriptions. The system covers data preprocessing
Optimizes model parameters to predict sentence descriptions by associating image features with ground-truth text.
Dieses Projekt ist ein umfassendes Bildungsprogramm und Deep-Learning-Framework, das darauf ausgelegt ist, praktisches Deep Learning mit PyTorch anhand von Notebooks und Codebeispielen zu vermitteln. Es dient als High-Level-Bibliothek zum Erstellen, Trainieren und Bereitstellen neuronaler Netze und fungiert als Modell-Trainings-Orchestrator, der PyTorch-Modelle, Optimierer und Verlustfunktionen koordiniert. Das Projekt bietet spezialisierte Toolkits für Computer Vision, Natural Language Processing und die Vorverarbeitung tabellarischer Daten. Es zeichnet sich durch fortschrittliche Trainingskontrollen aus, wie z. B. diskriminative Lernraten, ein Zwei-Wege-Callback-System zur Anpassung der Trainingslogik und eine High-Level-Learner-Abstraktion, die die Geräteplatzierung und Trainingsschleifen automatisiert. Das Framework deckt ein breites Fähigkeitsspektrum ab, einschließlich der automatisierten Konstruktion von Datenpipelines, der Analyse von Modellarchitekturen und der Leistungsbewertung bei Klassifizierungs-, Regressions- und Segmentierungsaufgaben. Es enthält zudem Dienstprogramme für verteiltes Training über mehrere GPUs, Mixed-Precision-Training zur Speicheroptimierung und spezialisierte Unterstützung für medizinische Bilddaten. Das Projekt wird als eine Reihe von Jupyter Notebooks bereitgestellt.
Concatenates texts into a continuous stream and splits them into sequences for language model training.
Kolors ist eine Implementierung eines generativen Modells zur Synthese fotorealistischer Bilder aus natürlichsprachlichen Beschreibungen und visuellen Referenzen. Es nutzt ein Latent-Diffusion-Modell-Framework, um hochauflösende Bilder zu erzeugen, und arbeitet innerhalb eines komprimierten latenten Raums, um Effizienz und Qualität der Generierung zu verbessern. Das System fungiert als mehrsprachiger Bildgenerator, der Text-Prompts in verschiedenen Sprachen interpretiert, um semantisch präzise visuelle Ergebnisse zu liefern. Es enthält eine benutzerdefinierte Modell-Trainings-Pipeline, die Low-Rank Adaptation (LoRA) verwendet, um dem Modell spezifische Motive oder künstlerische Stile anhand einer kleinen Anzahl von Bildern beizubringen. Das Projekt deckt ein breites Spektrum an Bildsynthese- und Bearbeitungsfunktionen ab, einschließlich Text-zu-Bild- und Bild-zu-Bild-Transformationen. Es bietet Tools zur Steuerung des räumlichen Layouts mittels Tiefen- oder Posenkarten, zur Injektion visueller Identität für ästhetische Konsistenz sowie maskenbasiertes Inpainting zur Rekonstruktion oder Modifikation spezifischer Bildbereiche. Die Implementierung enthält Dienstprogramme zur Bewertung der Bildqualität, die generierte Bilder anhand menschlicher Präferenzmetriken für ästhetische und semantische Qualität bewerten.
Provides the ability to interpret text prompts in multiple languages to produce semantically accurate visual outputs.
Dieses Projekt ist ein multimodaler Modell-Trainer und ein Tool für das Fine-Tuning von maschinellem Lernen, das einen containerisierten Workflow zur Anpassung vortrainierter Modelle an spezifische Aufgaben bietet. Es verfügt über eine No-Code-Weboberfläche und ein Dashboard für das Training von Large Language Models und anderen Datensätzen für maschinelles Lernen, ohne Code schreiben zu müssen. Das System zeichnet sich durch die Integration einer No-Code-Oberfläche mit Remote-GPU-Orchestrierung aus, die es Nutzern ermöglicht, containerisierte Trainingsumgebungen in der Cloud-Infrastruktur oder auf lokaler Hardware bereitzustellen. Es enthält einen dedizierten Integrator zum Hochladen trainierter Modellgewichte und Konfigurationen direkt in den Hugging Face Hub. Die Funktionen decken eine breite Palette an Modalitäten ab, einschließlich Computer Vision für Bildklassifizierung und Objekterkennung, Natural Language Processing für Textklassifizierung und Sequence-to-Sequence-Aufgaben sowie überwachtes Lernen für tabellarische Daten. Das Toolkit unterstützt zudem die Optimierung spezialisierter Sprachmodelle durch parameter-effizientes Fine-Tuning und Quantisierung. Trainings-Workflows werden über Konfigurationsdateien oder eine browserbasierte Oberfläche verwaltet, mit integrierter Unterstützung für das Mapping von Datensatzspalten und die Überwachung des Trainingsfortschritts.
Develops models that assign predefined categories or labels to blocks of text.
Spark NLP is a toolkit for scalable text analysis and machine learning built on the Apache Spark distributed computing framework. It provides a multimodal machine learning framework and a distributed pipeline system for sequencing annotators to process large-scale linguistic data. The library includes a transformer text processor for generating contextual vector embeddings and a dedicated inference engine for managing large language models. The project distinguishes itself through its ability to process heterogeneous data types, including text, audio, and images, within a unified vision-langu
Implements deep learning training for single-label multi-class and multi-label text categorization.
Dieses Projekt ist eine Implementierung der ALBERT-Sprachmodellarchitektur und bietet ein Framework zum Trainieren und Evaluieren von Transformer-basierten Textklassifikatoren und Ähnlichkeitsmodellen. Es enthält spezifisch vortrainierte Assets und Tools, die für die Generierung semantischer Embeddings und Repräsentationen chinesischer Texte optimiert sind. Das Framework zeichnet sich durch Tools zur Konvertierung schwerer Sprachmodell-Checkpoints in leichtgewichtige Formate aus, um Inferenz mit geringer Latenz auf Mobilgeräten zu ermöglichen. Es nutzt spezifische Techniken zur Gewichtsreduktion, einschließlich Cross-Parameter-Sharing und faktorisierter Embedding-Parametrisierung, um die Leistung bei geringerem Speicherbedarf aufrechtzuerhalten. Das System deckt eine vollständige Pipeline für Natural Language Processing ab, von der Normalisierung roher Texte und Subword-Tokenisierung bis hin zum selbstüberwachten Pre-training mittels Masked Language Modeling. Es bietet Funktionen für die Anpassung an nachgelagerte Aufgaben, wodurch vortrainierte Modelle für Textähnlichkeitsanalysen und überwachte Klassifizierungen feinabgestimmt werden können. Das Projekt enthält Dienstprogramme für die Konvertierung von Binärdatensätzen und die Transformation von Modellformaten, um die Kompatibilität über verschiedene Machine-Learning-Plattformen hinweg sicherzustellen.
Provides a supervised pipeline to train and evaluate text classifiers on specific language datasets.
Dieses Projekt ist ein PyTorch-basiertes Framework und Toolkit für die Texterkennung in Szenen. Es bietet eine Deep-Learning-Pipeline zur Extraktion von Zeichen und Wörtern aus Bildern natürlicher Umgebungen und deckt den gesamten Prozess von der Trainingsdatenaufbereitung bis zur Modellvalidierung ab. Das Framework fungiert als standardisierter Benchmark zur Messung der Genauigkeit und Inferenzgeschwindigkeit von Texterkennungsmodellen. Es enthält Tools zur Berechnung der Erkennungsgenauigkeit und zur Messung der GPU-Verarbeitungszeit pro Bild, um die Modellleistung über konsistente Datensätze hinweg zu bewerten. Das System umfasst visuelle und sequentielle Verarbeitungsstufen und nutzt konvolutive Merkmalsextraktion sowie rekursive Sequenzmodellierung. Es enthält Data-Engineering-Utilities für die Text- und Indexkonvertierung sowie Batch-Level-Datenausgleich, um Datensatzverteilungen während des Trainings zu verwalten.
Provides a complete pipeline for training deep learning models to recognize text from scene images.