from fastai2.tabular.all import *

path = untar_data(URLs.ADULT_SAMPLE)
path.ls()

df = pd.read_csv(path/'adult.csv')
df.head()

dls = TabularDataLoaders.from_csv(path/'adult.csv', path=path, y_names="salary",
    cat_names = ['workclass', 'education', 'marital-status', 'occupation', 'relationship', 'race'],
    cont_names = ['age', 'fnlwgt', 'education-num'],
    procs = [Categorify, FillMissing, Normalize])

dls.show_batch()

learn = tabular_learner(dls, metrics=accuracy)

learn.fit_one_cycle(1)

learn.show_results()

learn.predict(df.iloc[0])

test_df = df.copy()
test_df.drop(['salary'], axis=1, inplace=True)
dl = learn.dls.test_dl(test_df)

learn.get_preds(dl=dl)