Skip to content

PyTrial: A Comprehensive Platform for Artificial Intelligence for Drug Development

License

Notifications You must be signed in to change notification settings

RyanWangZf/PyTrial

Repository files navigation

PyTrial

PyTrial: A Python Package for Artificial Intelligence in Drug Development

Doc Tutorial

Downloads License Python 3.7+ Sunlab

A series of clinical trial task solutions and resources that are offered by PyTrial.

💡 News

⭐ Features

PyTrial is designed for both ML researchers and medical practioners, which is featured for

🚀 Installation

  • Before install PyTrial, we need to install pytorch first. Please refer to install PyTorch for a version that suits your device.

  • Then, it is easy to install PyTrial from github source:

pip install git+https://github.com/RyanWangZf/pytrial.git@main

The package is tested on python==3.7.

We DO NOT recommend downloading from PyPI temporarily because PyTrial is undergoing development swiftly.

☀️ Philosophy

In PyTrial, performing a task boils down to three steps: load data -> define model -> fit and predict.

To minimize the efforts learning to use PyTrial, we keep a consistent user interface for all tasks all models, i.e.,

model.fit(train_data, val_data)

model.predict(test_data)

model.save_model(save_dir)

model.load_model(load_dir)

hence all tasks are defined the input and output. All we need to do is to prepare for the input following the protocol.

📖 Documentation

We provide the following tutorials to help users get started with our PyTrial. After go through all these chapters, you will become the expert in AI for clinical trials and are ready to explore the frontier of this field.

The full documentation is at PyTrial-docs.

The principle of PyTrial

Tutorials for each task

Additional utilities

😃 Citing

If you use PyTrial in a scientific publication, we would appreciate citations to:

@article{wang2023pytrial,
  title={PyTrial: A Comprehensive Platform for Artificial Intelligence for Drug Development},
  author={Wang, Zifeng and Theodorou, Brandon and Fu, Tianfan and Xiao, Cao and Sun, Jimeng},
  journal={arXiv preprint arXiv:2306.04018},
  year={2023},
}