kubrik/modelSanoApp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

lalala

Browse source
This commit is contained in:
Kubrik 2024-05-21 10:00:33 +02:00
parent 42bb0f7901
commit b7576c633f
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
scripts/modelSanoApp.py
View file

@ -63,12 +63,13 @@ plasm_ant_features = ['CA_plasm_pre', 'CA.G_plasm_pre', 'CA.S_plasm_pre', 'To
'VA.GG_plasm_post', 'VA.S_post', 'VA.GS_plasm_post', 'VA.SS_plasm_post',
'Total.VA_plasm_post']
def model_trainig_multipleTarget(X_train, X_test, X_val, y_train, y_test, y_val):
def model_trainig_multipleTarget(X_train, X_test, X_val, y_train, y_test, y_val, xgb = True, rf = True):
print(" ----------------- STARTING XGB ----------------- ")
start = time.time()
if
xgb_model = XGBfit(X_train, X_test, X_val, y_train, y_test, y_val)
rf_model = RFfit(X_train, X_test, X_val, y_train, y_test, y_val)