Skip to content

Flyvannn/UrlDetection

Repository files navigation

URLS-Classification-Pytorch

##说明 model.py中包括多种模型,LSTM、LSTM+CNN、LSTM+CNN+Attn等
model4urls.py 采用以上模型实现URL分类,包括训练和测试过程
predict.py 以上模型预测代码
test.py 以上模型测试代码
bert4urlcls.py 采用BERT实现URL分类,包括训练和测试过程
multi_task.py 多任务训练
pretrain_and_ft.py 预训练+微调
test_xx.py 杂七杂八

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages