INNER CODE UNIT · Python

search

ShawhinT/YouTube-Blog · full-stack-data-science/ml-engineering/app/main.py:36

def search(query: str):
    idx_result = returnSearchResultIndexes(query, df, model, dist)
    return df.select(['title', 'video_id']).collect()[idx_result].to_dict(as_series=False)

View source record →