kipoi / models

Model zoo for genomics

Home Page:http://kipoi.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Fine-tune Basset from kipoi/models on a new dataset?

yding9 opened this issue · comments

Hello Kipoi members,

I want to use Basset from kipoi/models to fine-tune on a new dataset but I could not find the script. Could you provide the model fine-tune code?

Thank you so much in advance! Any help is highly appreciated!

Hi sorry for the late response:
Here is an example how to retrain/finetune a keras model:
https://kipoi.org/docs/tutorials/python-api/#re-train-the-keras-model and
https://gist.github.com/fchollet/7eb39b44eb9e16e59632d25fb3119975
For Basset, which is uses a PyTorch model, similar code is needed.
Note that kipoi does not provide an API to train/retrain/finetune models, therefore you have to use the underlying pytorch model directly.