Skip to content
Snippets Groups Projects
Commit 31ba5b48 authored by BARTHES Nicolas's avatar BARTHES Nicolas
Browse files

3-prediction.py : locked feature to the one of the model

parent 5b74352e
No related branches found
No related tags found
No related merge requests found
...@@ -142,7 +142,9 @@ with c2: ...@@ -142,7 +142,9 @@ with c2:
chem_data, spectra, meta_data, _ = dx_loader(change = hash_) chem_data, spectra, meta_data, _ = dx_loader(change = hash_)
st.success("The data have been loaded successfully", icon="") st.success("The data have been loaded successfully", icon="")
if chem_data.to_numpy().shape[1]>0: if chem_data.to_numpy().shape[1]>0:
yname = st.selectbox('Select target', options=chem_data.columns, index=chem_data.columns.to_list().index(system_data['data']['target'].name)) # yname = st.selectbox('Select target', options=chem_data.columns, index=chem_data.columns.to_list().index(system_data['data']['target'].name))
yname = system_data['data']['target'].name
st.info("Loaded model to predict " + yname)
measured = chem_data.loc[:,yname] == 0 measured = chem_data.loc[:,yname] == 0
y = chem_data.loc[:,yname].loc[measured] y = chem_data.loc[:,yname].loc[measured]
pred_data = spectra.loc[measured] pred_data = spectra.loc[measured]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment