def METHOD_NAME(args, predictor, val_df, test_df):
    val_features = extract_embedding(args, val_df, predictor, args.column_names)
    test_features = extract_embedding(args, test_df, predictor, args.column_names)
    val_labels = torch.tensor(val_df[args.label_column]).cuda()
    test_labels = torch.tensor(test_df[args.label_column]).cuda()
    return val_features, val_labels, test_features, test_labels