不知道該叫什么——泰坦尼克號個人項目代碼

寫在前面

這一點點代碼是分享會第一次留項目后我的作業(yè)代碼,留此備記,也方便為其他同學(xué)的Adaboost提供弱分類器。

代碼

# coding=utf-8
def predictions_qiu(data):
    """ Model with multiple features. Makes a prediction with an accuracy of at least 80%. """
    
    predictions = []
    for _, passenger in data.iterrows():
        if passenger[3] == 'female':
            if passenger[1] <= 2:  # female, Pclass 1、2. OVER
                predictions.append(1)
            else:  # female, Pclass 3
                if passenger[8] > 23.25:  # female, Pclass 3, Fare > 23.25
                    predictions.append(0)
                else:  # female, Pclass 3, Fare <= 23.25
                    if passenger[10] == 'Q':  # female, Pclass 3, Fare <= 23.25, Embarked =Q
                        if passenger[6] > 0:  # female, Pclass 3, Fare <= 23.25, Embarked =Q, Parch > 0
                            predictions.append(0)
                        else:  # female, Pclass 3, Fare <= 23.25, Embarked =C, Parch <= 0
                            if passenger[8] <= 7.65:
                                predictions.append(0)
                            else:
                                predictions.append(1)
                    elif passenger[10] == 'C':  # female, Pclass 3, Fare <= 23.25, Embarked=C
                        if passenger[8] > 15.246:
                            predictions.append(1)
                        else:
                            if passenger[8] > 13.863:
                                predictions.append(0)
                            else:
                                predictions.append(0)
                    else:  # female, Pclass 3, Fare <= 23.25, Embarked=S
                        if passenger[6] > 0:
                            predictions.append(1)
                        else:
                            predictions.append(0)
                
        else:  # male
            if passenger[4] > 10:  # # male, age > 10
                predictions.append(0)
            else:  # male, age < 10
                if passenger[5] > 2:  # male, age <= 10, SibSp > 2
                    predictions.append(0)
                else:  # male, age <= 10, SibSp <=2
                    if passenger[6] == 0:  # male, age <= 10, SibSp <=2, Parch == 0
                        predictions.append(0)
                    else:  # male, age <= 10, SibSp <=2, Parch > 0
                        predictions.append(1)
    return pd.Series(predictions)

我們的822,我們的青春
歡迎所有熱愛知識熱愛生活的朋友和822實驗室一起成長,吃喝玩樂,享受知識。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容