topicnet-topnet-integrated

TopicNet-TopNet Integrated Model

Unified Framework for Semantic Graph-Guided Topic Discovery and Neural Topic-Driven Story Generation

Model Architecture

• TopicNet: Semantic graph-guided topic discovery with LSTM encoder
• TopNet: Neural topic-driven story generation with cross-modal attention
• Cross-Modal Attention: Aligns topic discovery with story generation

Training Results

• Topic Loss: ~2.996
• Story Loss: ~9.211
• Trained on NVIDIA V100 GPU
• 20 epochs, 200 batches/epoch

Usage

from transformers import AutoModel, AutoTokenizer

model = AutoModel.from_pretrained("your-username/topicnet-topnet-integrated")
tokenizer = AutoTokenizer.from_pretrained("your-username/topicnet-topnet-integrated")

Citation

@article{topicnet_topnet_2024,
  title={Unified Framework for Semantic Graph-Guided Topic Discovery and Neural Topic-Driven Story Generation},
  author={Your Name},
  journal={arXiv preprint},
  year={2024}
}