Python Forum
Which network architecture should be chosen to predict disease severity? - Printable Version

+- Python Forum (https://python-forum.io)
+-- Forum: Python Coding (https://python-forum.io/forum-7.html)
+--- Forum: Data Science (https://python-forum.io/forum-44.html)
+--- Thread: Which network architecture should be chosen to predict disease severity? (/thread-34685.html)



Which network architecture should be chosen to predict disease severity? - AlekseyPython - Aug-20-2021

Python 3.9.6, Pytorch 1.9.0+cu111

I have several text fields describing various aspects of a patient's treatment (filled in by the doctors). The entire samples is about 20 thousand cases. It's necessary to answer whether the severity of the disease was more than the specified one (yes or no).

From the book, I took the following sequence of layers of a neural network (but there's a completely different task):

Embedding (64 dimensions)
LSTM, unidirectional (64 -> 32)
Linear (32 -> 16)
Sigmoid

There are few samples (~ 20 thousands), so I tried to reduce the complexity of the network as much as possible, so that it really learns, and not just remembers samples.

Is there any good literature that explains, which layer is best used in which situation (and in what sequence)? Can you give tips on the correct network architecture for this case?

Thank you in advance for participating in the conversation.


RE: Which network architecture should be chosen to predict disease severity? - Larz60+ - Aug-21-2021

you might possibly find something here: https://pypi.org/search/?q=%22clinical+trial%22
These are mainly related to clinical trials, but i would think there might be something that uses neural networks and is related to what your doing.
Then again, maybe not.


RE: Which network architecture should be chosen to predict disease severity? - jefsummers - Aug-21-2021

A couple sources of info that may help:
Choosing an activation function

More info on activation functions

Hope this helps