/HGHDA

Primary LanguageJupyter NotebookMIT LicenseMIT

HGHDA

Dual-channel hypergraph convolutional network for predicting herb-disease associations

📂Experimental Results

  1. Contains the results of the experiment and the results of the comparison algorithm such as
    • Randomly_shuffled_graph file is an experimental result for randomly shuffled graphs
    • c_tanimoto_sim file is the model result after calculating the effective component similarity using tanimoto coefficients
    • cold_start file is the result of a cold-start experiment
    • result file is the result of the model and comparison algorithm

📃Dataset

Original data can be viewed by clicking on zenodo here

🚀 Installation

Installation of the project runtime environment,First you can create a virtual environment for the project:

$ python -m venv [your env name]

Activate the virtual environment:

$ source [your env name]/bin/activate

Install project dependencies:

$ pip install -r requirements.txt

🏃 Running

Go to the project directory:

$ cd Project_Path/src

Run the main.py file in the src folder:

$ python main.py

🛠️ Configuration

The model can be configured via HGHDA.conf in the src folder

  • datapath:Set the file path of the dataset
  • ratings.setup:Defaults to -columns 0 1 2 (herb,disease,rating)
  • evaluation.setup:Folds for cross validation
  • num.factors:the number of latent factors
  • num.max.epoch:the maximum number of epoch for algorithms.
  • output.setup:the directory path of output results

🚅 Benchmarks

  1. For the baseline code of the comparison algorithm, we give the relevant links.
    • Please refer the code of BiGI here
    • Please refer the code of SMGCL here
    • Please refer the code of MilGNet here
    • Please refer the code of LHGCE here
    • Please refer the code of HGNNLDA here
  2. For the HTInet algorithm, we performed a simple replication of it, using herb-component, component-target, and target-disease association data in experiments. For SMGCL, we use the similarity of effective components to obtain the similarity of herbs and the similarity of targets to obtain the similarity of diseases as their inputs.
  3. With respect to the parameter settings involved in running these algorithms, we either explicitly adopted the default settings recommended by their publications or set their associated parameters to the same or similar values as HGHDA.

⚖️ License

The code in this package is licensed under the MIT License.