FTTransformer 遇到 ModuleNotFoundError: No module named 'omegaconf'

[GoldAndRabbit]

---

ModuleNotFoundError Traceback (most recent call last)
Cell In[5], line 1
----> 1 from torchkeras.tabular import TabularPreprocessor
2 from sklearn.preprocessing import OrdinalEncoder
4 #特征工程

File /Library/Frameworks/Python.framework/Versions/3.12/lib/python3.12/site-packages/torchkeras/tabular/init.py:1
----> 1 from . import models
2 from . import utils
3 from .utils import available_models

File /Library/Frameworks/Python.framework/Versions/3.12/lib/python3.12/site-packages/torchkeras/tabular/models/init.py:1
----> 1 from . import (
2 autoint,
3 category_embedding,
4 danet,
5 ft_transformer,
6 gandalf,
7 gate,
8 node,
9 tab_transformer,
10 tabnet
11 )
13 from .autoint import AutoIntConfig, AutoIntModel
...
----> 8 from omegaconf import DictConfig
10 from ..common.layers import Embedding2dLayer
11 from torchkeras.tabular.utils import _initialize_layers, _linear_dropout_bn

ModuleNotFoundError: No module named 'omegaconf'
Output is truncated. View as a scrollable element or open in a text editor. Adjust cell output settings...

请问有人知道什么原因吗

[lyhue1991]

pip install 一下