Python convey示例

示例#1

文件： iris.py 项目： yusugomori/conveyer

import os
from glob import glob
import conveyer

if __name__ == '__main__':
    '''
    Multiclass Classification Task
    '''
    data_dir = os.path.join(os.path.dirname(__file__), 'data', 'iris')
    f_train = os.path.join(data_dir, 'train.csv')

    # training and validation
    conveyer.convey(path=f_train, out_dir='model')

示例#2

文件： house_prices.py 项目： yusugomori/conveyer

import os
from glob import glob
import conveyer


if __name__ == '__main__':
    '''
    Regression Task
    Assume we have `data/train.csv` and `data/test.csv`
    from
    https://www.kaggle.com/c/house-prices-advanced-regression-techniques/data
    '''
    data_dir = os.path.join(os.path.dirname(__file__), 'data')
    f_train = os.path.join(data_dir, 'train.csv')
    f_test = os.path.join(data_dir, 'test.csv')

    # training and validation
    conveyer.convey(path=f_train,
                    ignore_cols=['Id'],
                    out_dir='model')

    # test
    preds = conveyer.produce(f_test, model_dir='model')
    print(preds)

示例#3

import os
from glob import glob
import conveyer

if __name__ == '__main__':
    '''
    Binary Classification Task
    Assume we have `data/train.csv` and `data/test.csv`
    from https://www.kaggle.com/c/titanic/data
    '''
    data_dir = os.path.join(os.path.dirname(__file__), 'data')
    f_train = os.path.join(data_dir, 'train.csv')
    f_test = os.path.join(data_dir, 'test.csv')

    # training and validation
    conveyer.convey(path=f_train,
                    ignore_cols=['PassengerId', 'Name'],
                    out_dir='model',
                    target_name='Survived')

    # test
    preds = conveyer.produce(f_test, model_dir='model')
    print(preds)