diff --git a/.gitignore b/.gitignore index 471b461..68ae4f2 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ # Hide node dependencies directory -node_modules \ No newline at end of file +node_modules +__pycache__ \ No newline at end of file diff --git a/src/products.tsv b/src/data/products.tsv similarity index 100% rename from src/products.tsv rename to src/data/products.tsv diff --git a/src/decision_tree/main.py b/src/decision_tree/main.py new file mode 100755 index 0000000..a686e5a --- /dev/null +++ b/src/decision_tree/main.py @@ -0,0 +1,10 @@ +#!/usr/bin/python3 +from sklearn import tree + +def read_tsv_from(filename): + from csv import reader + with open(filename, 'r') as f: + header, *rows = list(reader(f, delimiter='\t')) + return [dict(zip(header, row)) for row in rows] + +print(read_tsv_from('./data.tsv'))