The treatment of the image datasets in WideDeepDataset replicates that of Pytorch. In particular this source code:

if isinstance(pic, np.ndarray):
    # handle numpy array
    if pic.ndim == 2:
        pic = pic[:, :, None]

In addition, I have added the possibility of using each of the model components in isolation and independently. This is, one could now use the wide, deepdense (either DeepDense or DeepDenseResnet), deeptext and deepimage independently.

项目简介

当前项目暂无项目简介

发行版本 21

Flash and Linear Attention mechanisms added to the TabTransformer

全部发行版

贡献者 10

开发语言

  • Python 97.2 %
  • JavaScript 2.4 %
  • CSS 0.4 %
  • Shell 0.0 %