Commit d7691eb0 authored by Christian Marius Lillelund's avatar Christian Marius Lillelund
Browse files

added db script, retrained models

parent 06de8271
kmodes==0.11.0
matplotlib==3.3.4
numpy==1.19.5
pandas==1.2.3
pytest==6.2.2
pytest-runner==5.3.0
scikit-learn==0.24.1
Sphinx==3.5.2
notebook==6.2.0
streamlit==0.78.0
tensorflow==2.4.1
openpyxl==3.0.6
xgboost==1.3.3
keras-tuner==1.0.2
shap==0.37.0
scikeras==0.2.1
fastapi==0.63.0
uvicorn==0.13.4
fastapi-jwt-auth==0.5.0
\ No newline at end of file
......@@ -26,7 +26,7 @@ def main():
labels_enc[col_name] = le
X.loc[:, col_name] = le.transform(X.loc[:, col_name].astype(str))
model = RandomSurvivalForest(n_estimators=200, max_depth=6,
model = RandomSurvivalForest(n_estimators=200, max_depth=3,
n_jobs=-1, random_state=0)
model.fit(X, y)
......
......@@ -23,6 +23,7 @@ REPORTS_PLOTS_DIR = Path.joinpath(ROOT_DIR, 'reports/plots')
TESTS_FILES_DIR = Path.joinpath(ROOT_DIR, 'tests/files')
RAW_DATA_DIR_2019 = Path.joinpath(ROOT_DIR, 'data/raw/2019')
RAW_DATA_DIR_2020 = Path.joinpath(ROOT_DIR, 'data/raw/2020')
RAW_DATA_DIR_TEST = Path.joinpath(ROOT_DIR, 'data/raw/test')
PROCESSED_DATA_DIR = Path.joinpath(ROOT_DIR, 'data/processed')
PROCESSED_EMB_DATA_DIR = Path.joinpath(ROOT_DIR, 'data/processed/embeddings')
INTERIM_DATA_DIR = Path.joinpath(ROOT_DIR, 'data/interim')
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment