Training Model Random Forest dengan Data ‘smsspamcollection’

[rohanaziz1811]

halo coach, aku mencoba untuk train model random forest dengan dataset ‘smsspamcollection’, saat fitting model, ada eror:
could not convert string to float: ‘ham’
solusinya gimana ya coach?

aku lampirkan script ku :
import numpy as np
import pandas as pd
df = pd.read_csv('smsspamcollection.tsv', sep ='\t')

#train random forest
from sklearn.ensemble import RandomForestRegressor

rf_model = RandomForestRegressor(n_estimators = 1000, random_state = 42)
rf_model.fit(X_train,y_train)

[Coach William]

Hi Rohana,

Jika dilihat dr error messagenya, fungsinya tdk bs menerima target berupa string yaitu ham dan spam. Mungkin kamu bisa coba ubah jadi 0 dan 1 sebelum dimasukan ke fungsi nya.

Regards

[rohanaziz1811]

untuk solusinya, apakah bisa menggunakan onehotencoder coach?

[Coach William]

Hi Rohana,

Sori saya baru sadar kamu menggunakan randomforestregressor,
Coba ganti pakai randomforestclassifier ya

>>> from sklearn.ensemble import RandomForestClassifier

>>> clf = RandomForestClassifier(max_depth=2, random_state=0)
>>> clf.fit(X, y)

Lengkapnya di
https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html#sklearn.ensemble.RandomForestClassifier.fit

[Author]

Posts