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

split feature making and target making

parent d1d74131
......@@ -81,7 +81,7 @@ def main():
print(f"Completed making models. Models and SHAP plots can be found at:\n" +
f"{pt.COMPLETE_XGB_DIR}\n" + f"{pt.COMPLIANCE_XGB_DIR}\n" +
f"{pt.FALL_XGB_DIR}\n" + f"{pt.FALL_TEST_XGB_DIR}" + "\n")
f"{pt.FALL_XGB_DIR}\n" + f"{pt.RISK_XGB_DIR}" + "\n")
if __name__ == "__main__":
main()
\ No newline at end of file
......@@ -8,7 +8,7 @@ import joblib
from pathlib import Path
def main():
model = file_reader.read_joblib(pt.FALL_TEST_XGB_DIR,
model = file_reader.read_joblib(pt.RISK_XGB_DIR,
'fall_test_xgboost.joblib')
for gender in range(0, 2):
......
......@@ -8,7 +8,7 @@ import joblib
from pathlib import Path
def main():
model = file_reader.read_joblib(pt.FALL_TEST_XGB_DIR,
model = file_reader.read_joblib(pt.RISK_XGB_DIR,
'fall_test_xgboost.joblib')
for gender in range(0, 2):
......
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