A basic model that can be used on tabular data
from nbdev.showdoc import *

Model

emb_sz_rule[source]

emb_sz_rule(n_cat)

Rule of thumb to pick embedding size corresponding to n_cat

get_emb_sz[source]

get_emb_sz(to, sz_dict=None)

Get default embedding size from TabularPreprocessor proc or the ones in sz_dict

class TabularModel[source]

TabularModel(emb_szs, n_cont, out_sz, layers, ps=None, embed_p=0.0, y_range=None, use_bn=True, bn_final=False, bn_cont=True) :: Module

Basic model for tabular data.

tabular_config[source]

tabular_config(ps=None, embed_p=0.0, y_range=None, use_bn=True, bn_final=False, bn_cont=True)

Convenience function to easily create a config for tabular_model