search

gongtiansu / KnotFold

Accurate Prediction of RNA Secondary Structure Including Pseudoknots through Solving Minimum-cost Flow with Learned Potential
MIT License
3 stars 2 forks source link

Dataset preparation for training #3

Open skoc opened 4 months ago

skoc commented 4 months ago

Hi, thank you for great work, and sharing the code!

Can you share your Dataset preparation functions used in the training?

self.dataset = RNADataset(train=True) self.dataset = BatchDataset( self.dataset, {"seq": 0, "contact": -1, "pseudoknot": -1} )