Skip to main content

this is a feature column and transformer tools in pytorch, which is like the featureColumn class in Tensorflow

Project description

Feature Column in Pytorch


this is a feature column tool implemented by pytorch, which has the same function compared to featurecol class in Tensorflow.

And it's also a tool for me to paricipate in some data science competition.


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
  • Example
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=[])
    res = Features_1.transform(data)

Returned Data

# [<number_feat.Number object at 0x0000018903E64748>]
# OrderedDict([('col1', array([-1.4048787 ,  0.8429272 ,  0.56195146], dtype=float32)), ('col3', array([1, 1, 1], dtype=int64))])

Project details

Release history Release notifications | RSS feed

This version


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for featureColByTorch, version 1.0
Filename, size File type Python version Upload date Hashes
Filename, size featureColByTorch-1.0-py2.py3-none-any.whl (5.9 kB) File type Wheel Python version py2.py3 Upload date Hashes View
Filename, size featureColByTorch-1.0.tar.gz (3.5 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page