Using an Autoencoder to encode features for k-Means Clustering on the AG News Dataset
-
Updated
Jul 9, 2021 - Python
Using an Autoencoder to encode features for k-Means Clustering on the AG News Dataset
The objective of this task is to build a text classification model using the Hugging Face library to classify a dataset of text into one of multiple categories.
This project fine-tunes the DistilBERT model on the AG News dataset for text classification
A text classification project combining attention mechanisms with recurrent models (LSTM and GRU) on the AG News dataset using TensorFlow.
This repository contains the code, model configurations, and report for fine-tuning a roberta-base model using Low-Rank Adaptation (LoRA)- a Parameter-Efficient Fine-Tuning (PEFT) method - on the AG News classification task. The goal was to achieve high test accuracy while keeping trainable parameters under 1 million.
Predicting the topic of news articles
Add a description, image, and links to the ag-news-dataset topic page so that developers can more easily learn about it.
To associate your repository with the ag-news-dataset topic, visit your repo's landing page and select "manage topics."