INNER CODE UNIT · Python

retrieve_dataset

featurestoreorg/serverless-ml-course · src/04-module/cc-fraud-streamlit-ui.py:35

def retrieve_dataset(fv, start_date, end_date):
    st.write(36 * "-")
    print_fancy_header('\n💾 Dataset Retrieving...')
    batch_data = fv.get_batch_data(start_time = start_date, end_time = end_date)
    batch_data.drop(["tid", "cc_num", "datetime"], axis = 1, inplace=True)
    return batch_data


@st.cache(suppress_st_warning=True, allow_output_mutation=True)
def get_feature_view():
    fv = fs.get_feature_view("cc_trans_fraud", 1)
    return fv


@st.cache(allow_output_mutation=True,suppress_st_warning=True)
def get_model(project = project):
    mr = project.get_model_registry()
    model = mr.get_model("cc_fraud", version = 1)

View source record →