# MacBERT

**Repository Path**: chenyaopeng/mac-bert

## Basic Information

- **Project Name**: MacBERT
- **Description**: 本项目利用MacBERT预训练模型对新闻标题文本数据进行分类，分类类型包含楼市类、股市类、教育类，并根据新的新闻标题预测其所属分类。
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2024-12-08
- **Last Updated**: 2024-12-08

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## 简介

本项目利用MacBERT预训练模型对新闻标题文本数据进行分类，分类类型包含楼市类、股市类、教育类，并根据新的新闻标题预测其所属分类。

## 什么是MacBERT

本目录包含**MacBERT预训练模型**，该模型引入了一种纠错型掩码语言模型（Mac）预训练任务，缓解了“预训练-下游任务”不一致的问题。MacBERT在多种NLP任务上取得了显著性能提升。

项目地址 https://github.com/ymcui/MacBERT

## train.py 训练

![img.png](img.png)

## main.py 预测

![img_1.png](img_1.png)