밑바닥 부터 시작하는 머신러닝 입문 > Chapter 4 문제 맞게 풀었는지 확인이 필요합니다.

안녕하세요.

처음하는 머신러닝이라 문제를 풀어서 통과를 했는데, 맞게 한지 모르겠습니다.

아래처럼 하면 되는지 확인 부탁드립니다.

import numpy as np

import pandas as pd

from pandas import DataFrame

def get_rating_matrix(filename, dtype=np.float32):

df = pd.read_csv(filename)

user = sorted(df.source.unique())

movie = sorted(df.target.unique())

data = {mv: {us[0]: us[1] for us in df[["source","rating"]][df["target"] == mv].values} for mv in movie}

df_data = DataFrame(data, index=user).fillna(0)

result = df_data.values

return result

def get_frequent_matrix(filename, dtype=np.float32):

df = pd.read_csv(filename)

index_name = sorted(df.source.unique())

columns_name = sorted(df.target.unique())

data = { idx : { t[0]:t[1] for t in zip(df["target"][df["source"] == idx].value_counts().index,  df["target"][df["source"] == idx].value_counts().values) } for idx in index_name }
df2 = DataFrame(data, dtype=np.float32)
result = df2.T.values
return result