python transformer category_test.py · tianyu/featureColByTorch - Gitee.com

#!/usr/bin/env python

# -*- encoding: utf-8 -*-

"""

@File : test.py

@Author : Braylon1002

@Version : 1.0

@Contact : S.Braylon1002@gmail.com

Desc : None

"""

from features import Features

from number_feat import Number

from category_feat import Category

from transformers.NumberTransformer import NumberTransformer

from transformers.CategoryTransformer import CategoryTransformer

import pandas as pd

import numpy as np

def make_data():

data = pd.DataFrame(np.random.randint(1, 20, size=(3,4)), columns=['col1', 'col2', 'col3', 'col4'])

return data

if __name__ == "__main__":

data = make_data()

Number_1 = Number('col1', NumberTransformer())

Category_1 = Category('col2', CategoryTransformer())

Category_2 = Category('col3', CategoryTransformer())

Features_1 = Features(number_feat=[Number_1], category_feat=[Category_2], sequence_feat=[])

print(Features_1.number_feat)

Features_1.fit(data)

res = Features_1.transform(data)

print(res)

一键复制

编辑

Web IDE

原始数据

按行查看

历史