Skip to content

Commit 9c478ef

Browse files
authored
Update README.md
1 parent 911ade4 commit 9c478ef

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,18 @@ Authored and maintained by **Dr. Tirthajyoti Sarkar ([Website](https://tirthajyo
3434
3535
---
3636

37-
## Utility function
37+
## Utility modules
3838

39+
### Utility module for example notebooks
3940
I created a utility function file called `DL_utils.py` in the `utils` directory under `Notebooks`. We use functions from this module whenever possible in the Jupyter notebooks.
4041

4142
You can download the module (raw Python file) from here: [DL-Utility-Module](https://raw.githubusercontent.com/tirthajyoti/Deep-learning-with-Python/master/Notebooks/utils/DL_utils.py)
4243

44+
### General-purpose regression module (for tabular dataset)
45+
I also implemented a general-purpose trainer module (`NN_trainer.py`) for regression task with tabular datasets. The idea is that you can simply read a dataset (e.g. a CSV file), choose the input and target variables, build a densely-connected neural net, train, and predict. The module gives you back a prediction function (trained) which can be used for any further prediction, analytics, or optimization task.
46+
47+
Check out the module [here](https://github.com/tirthajyoti/Deep-learning-with-Python/blob/master/utils/NN_trainer.py) and an example notebook here.
48+
4349
## Notebooks
4450

4551
### Deep learning vs. linear model

0 commit comments

Comments
 (0)