csinva / imodelsX

Scikit-learn friendly library to interpret, and prompt-engineer text datasets using large language models.

Home Page:https://csinva.io/imodelsX/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Does this implementation accept multiple hidden layers?

valeman opened this issue · comments

I was wondering if this implementation allowed for multiple hidden layers that is key in getting good performance.

I just pushed support for this :)

You'll need to update the package (pip install imodelsx), then can pass hidden_layer_sizes list for multiple layers (like sklearn's MLP). See the second cell of the updated notebook for an example.