인프런 커뮤니티 질문&답변

섹션14 작업형2 어떤 부분이 문제일까요 ㅠㅠ?

ValueError: could not convert string to float: 'Healthcare'

이런 오류가 뜹니다..

target = train.pop('Segmentation')

test_ID = test.pop('ID')

# 연속형 변수 스케일링

num = ['Age', 'Work_Experience', 'Family_Size']

from sklearn.preprocessing import RobustScaler

scaler = RobustScaler()

train[num] = scaler.fit_transform(train[num])

test[num] = scaler.fit_transform(test[num])

# 범주형 변수 원핫인코딩

cols = ['Gender', 'Ever_Married', 'Graduated', 'Spending_Score']

train = pd.get_dummies(data = train, columns = cols)

test = pd.get_dummies(data = test, columns = cols)

# 평가

from sklearn.ensemble import RandomForestClassifier

model = RandomForestClassifier()

model.fit(train, target)

pred = model.predict(test)

# 제출

pred = model.predict(test)

pd.DataFrame({'ID': test_cust_id, 'Segmentation': pred}).to_csv('003000000.csv', index=False)