Laborratte5/sci-kit-classify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Parse cli arguments using argparse

Browse source
This commit is contained in:
Laborratte 5 2025-06-22 18:00:20 +02:00
parent 0bf7c19f86
commit 72709651fa
Signed by: Laborratte5
GPG key ID: 3A30072E35202C02
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/cluster_test/cluster.py
View file

@ -1,3 +1,4 @@
import argparse
import numpy as np import numpy as np
import pandas import pandas
from sklearn.impute import SimpleImputer from sklearn.impute import SimpleImputer
@ -32,8 +33,18 @@ def classify(X, data_frame):
return data_frame[y_pred < 0] return data_frame[y_pred < 0]
if __name__ == '__main__': if __name__ == '__main__':
data_frame = read_excel("tests/Beispiel Auswertung2.xlsx") parser = argparse.ArgumentParser()
X = pandas.get_dummies(data_frame) # OneHotEncode categorical values parser.add_argument("file")
parser.add_argument(
"--data-column",
default="Messpunkt",
help="Used to identify the columns containing measurements. The heading of each column containing data should contain the specified value."
)
args = parser.parse_args()
data_frame = read_excel(args.file)
X = data_frame.filter(like=args.data_column, axis='columns')
X = pandas.get_dummies(X) # OneHotEncode categorical values
pipe = make_pipeline(SimpleImputer(add_indicator=True)) pipe = make_pipeline(SimpleImputer(add_indicator=True))
X = pipe.fit_transform(X) X = pipe.fit_transform(X)
outlier = classify(X, data_frame) outlier = classify(X, data_frame)