Brane Scikit-learn package

Requirements

pyyaml==5.4.1

pandas==1.2.4

numpy==1.20.3

sklearn==1.1.1

flake8

Installation

brane import ZNBai/Compute

Functions

NAME	INPUT EXAMPLES	OUTPUT EXAMPLES
clean	INPUT: '/data/train.csv' OUTPUT_PATH: '/data/'	/data/clean.csv
numeralization	INPUT: '/data/train.csv' OUTPUT_PATH: '/data/'	/data/numeralization.csv
normalization	INPUT: '/data/train.csv' OUTPUT_PATH: '/data/'	/data/normalization.csv
train	INPUT: '/data/train.csv' OUTPUT_PATH: '/data/'	clf.pickle
test	INPUT: '/data/test.csv' MODEL_PATH: '/data/' OUTPUT_PATH: '/data/'	/data/prediction.csv

Tests

Run unit tests with pytest: pytest