Transaction Prediction

Transaction Prediction Model

Function get_data():

Returns pandas.dataframe, pandas.dataframe

Usage

    data_train, data_test = get_data(train = 'https://raw.githubusercontent.com/../transaction_dataset_train.csv', test = 'https://raw.githubusercontent.com/../transaction_dataset_test.csv')

Function preprocess_inputs():

Returns pandas.dataframe, pandas.dataframe

Usage

    X, y = preprocess_inputs(data_train, data_test, "Random_Forest")

Function pretrained():

Returns model

Usage

    model = pretrained("Random_Forest")

Function train():

Returns model

Usage

    model = train(data_train, data_test, "Random_Forest")

Function predict():

Returns array

Usage

    print(predict(data_test, model))

Last updated