imdb_success_predictor.py 文件源码

def create_input(movie_info):
    # don't want to cinlude movie_id, title, country in predicition
    SKIP = 3
    WIDTH = len(movie_info[0]) - SKIP
    X = scipy.zeros((len(movie_info), WIDTH))
    for i in range(0, len(movie_info)):
        for j in range(SKIP, WIDTH):
        try:
                    X[i, j-SKIP] = movie_info[i][j] if movie_info[i][j] != '' else 0
        except Exception:
            pass
    return X