Doctorado-ML/fimdlp
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/FImdlp.git synced 2025-08-16 16:05:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add mdlp version to Makefile

Browse Source 
Refactor sample.py
This commit is contained in:
Ricardo Montañana Gómez
2023-04-25 17:11:40 +02:00
parent 17a66858f8
commit 3a100bbba7
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Makefile
View File

@@ -40,6 +40,7 @@ audit: ## Audit pip
version: version:
@echo "Current Python version .: $(shell python --version)" @echo "Current Python version .: $(shell python --version)"
@echo "Current FImdlp version .: $(shell python -c "from fimdlp import _version; print(_version.__version__)")" @echo "Current FImdlp version .: $(shell python -c "from fimdlp import _version; print(_version.__version__)")"
@echo "Current mdlp version ...: $(shell python -c "from fimdlp.cppfimdlp import CFImdlp; print(CFImdlp().get_version().decode())")"
@echo "Installed FImdlp version: $(shell pip show fimdlp | grep Version | cut -d' ' -f2)" @echo "Installed FImdlp version: $(shell pip show fimdlp | grep Version | cut -d' ' -f2)"
help: ## Show help message help: ## Show help message

17
samples/sample.py
View File

@@ -1,7 +1,6 @@
import time import time
import argparse import argparse
import os import os
from scipy.io import arff
import pandas as pd import pandas as pd
from sklearn.ensemble import RandomForestClassifier from sklearn.ensemble import RandomForestClassifier
from fimdlp.mdlp import FImdlp from fimdlp.mdlp import FImdlp
@@ -18,11 +17,21 @@ datasets = {
ap = argparse.ArgumentParser() ap = argparse.ArgumentParser()
ap.add_argument( ap.add_argument(
"--min_length", type=int, default=3, help="Minimum length of interval" "-n",
"--min_length",
type=int,
default=3,
help="Minimum length of interval",
) )
ap.add_argument("--max_depth", type=int, default=9999, help="Maximum depth")
ap.add_argument( ap.add_argument(
"--max_cuts", type=float, default=0, help="Maximum number of cut points" "-m", "--max_depth", type=int, default=9999, help="Maximum depth"
)
ap.add_argument(
"-c",
"--max_cuts",
type=float,
default=0,
help="Maximum number of cut points",
) )
ap.add_argument("dataset", type=str, choices=datasets.keys()) ap.add_argument("dataset", type=str, choices=datasets.keys())
args = ap.parse_args() args = ap.parse_args()