search

cannin / graph_neural_network_drug_response

2 stars 0 forks source link

Train Benchmark AutoML Model #7

Open cannin opened 2 years ago

cannin commented 2 years ago

Use the https://autokeras.com/ AutoML system as a starting point.

inoue0426 commented 2 years ago

I tried to use autokeras. Colab You can check the result from this link.

inoue0426 commented 2 years ago

I tried to use cell2mutation and morgan fingerprinting as features, but this occurred out of memory. Cell2mutation is 3008 dim and fingerprinting is 2048 dim and total matrix dimension was 100K * 5k, and this caused OOM. So, I'll try to make small sample dataset and then try to use autokeras and DrugCell.