コンペにあるタイタニックの生存を予測する課題にlightgbm(trainAPI)
を使用しました。
paramus内の設定で下記の様に'objective': 'binary'としましたが
pre = model2.predict(test)のpreが0,1では無く
array([0.1095665 , 0.85344621, 0.49506048, 0.11741223, 0.2672058 ,
0.49506048, 0.49506048, 0.15292294, 0.49506048, 0.75977963,
0.12123925, 0.49506048, 0.12239546, 0.12084048, 0.53556324,・・・
となります。
何故'objective': 'binary'なのに0,1の結果がに成らないのでしょうか?
またどの様にすれば望みの結果となるのか教えて頂けますでしょうか?
コーディングは一部を抜粋(タイタニックの生存予測です)
train_data = lgb.Dataset(trainx,trainy) #trainデータを7:3に分割しました。
val_data= lgb.Dataset(valx,valy)
params = {
'objective': 'binary',
'num_leaves': 50,
'min_data_in_leaf': 30,
'max_depth': 5,
}
model2 = lgb.train(
params=params,
train_set=train_data,
num_boost_round=1000,
early_stopping_rounds=100,
valid_sets=val_data)
pre = model2.predict(test)
pre
array([0.1095665 , 0.85344621, 0.49506048, 0.11741223, 0.2672058 ,
0.49506048, 0.49506048, 0.15292294, 0.49506048, 0.75977963,
0.12123925, 0.49506048, 0.12239546, 0.12084048, 0.53556324,
0.27255102, 0.35212627, 0.85344621, 0.35212627, 0.2672058 ,
0.1095665 , 0.49506048, 0.53556324, 0.49506048, 0.11365671
的外れな質問かもしれませんが初心者なものでお許しください。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2021/12/29 00:40