Senasu commited on
Commit
dba52bd
Β·
verified Β·
1 Parent(s): 4af88ff

Upload 4 files

Browse files
Files changed (4) hide show
  1. app.py +64 -0
  2. best_regression_model.pkl +3 -0
  3. requirements.txt +3 -0
  4. train.csv +0 -0
app.py ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ import joblib
4
+ from sklearn.pipeline import Pipeline
5
+ from sklearn.compose import ColumnTransformer
6
+ from sklearn.preprocessing import StandardScaler
7
+
8
+ # Load data and model
9
+ st.set_page_config(page_title="Mohs Hardness Prediction", layout="centered")
10
+
11
+ df = pd.read_csv("train.csv")
12
+ model = joblib.load("best_regression_model.pkl")
13
+
14
+ # Feature list
15
+ FEATURES = ['allelectrons_Total', 'density_Total', 'allelectrons_Average',
16
+ 'val_e_Average', 'atomicweight_Average', 'ionenergy_Average',
17
+ 'el_neg_chi_Average', 'R_vdw_element_Average', 'R_cov_element_Average',
18
+ 'zaratio_Average', 'density_Average']
19
+
20
+ # Create pipeline
21
+ preprocessor = ColumnTransformer([
22
+ ("num", StandardScaler(), FEATURES)
23
+ ])
24
+ pipeline = Pipeline(steps=[("preprocessor", preprocessor), ("regressor", model)])
25
+ pipeline.fit(df[FEATURES], df["Hardness"])
26
+
27
+ def hardness_prediction(input_data):
28
+ prediction = pipeline.predict(pd.DataFrame([input_data]))[0]
29
+ return float(prediction)
30
+
31
+ # Application title
32
+ st.title("πŸ” Mohs Hardness Prediction")
33
+ st.markdown("Predict the Mohs hardness of a material based on its properties.")
34
+
35
+ # User input form
36
+ with st.form("prediction_form"):
37
+ st.subheader("πŸ“Š Model Inputs")
38
+ col1, col2 = st.columns(2)
39
+
40
+ inputs = {}
41
+ input_params = [
42
+ ("allelectrons_Total", 0, 20000, 100),
43
+ ("density_Total", 0, 10000, 50),
44
+ ("allelectrons_Average", 0, 100, 1),
45
+ ("val_e_Average", 0.0, 10.0, 0.1),
46
+ ("atomicweight_Average", 0, 200, 1),
47
+ ("ionenergy_Average", 0, 100, 1),
48
+ ("el_neg_chi_Average", 0.0, 10.0, 0.1),
49
+ ("R_vdw_element_Average", 0.0, 5.0, 0.01),
50
+ ("R_cov_element_Average", 0.0, 5.0, 0.01),
51
+ ("zaratio_Average", 0.0, 1.0, 0.01),
52
+ ("density_Average", 0, 10, 1)
53
+ ]
54
+
55
+ for i, (feature, min_v, max_v, step_v) in enumerate(input_params):
56
+ col = col1 if i % 2 == 0 else col2 # Arrange inputs in two columns
57
+ inputs[feature] = col.number_input(feature, min_value=min_v, max_value=max_v, step=step_v)
58
+
59
+ submitted = st.form_submit_button("πŸš€ Predict")
60
+
61
+ # Show prediction result
62
+ if submitted:
63
+ prediction = hardness_prediction(inputs)
64
+ st.success(f"**Predicted Mohs Hardness: {prediction:.2f}**")
best_regression_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:495aeab26037706de59362d83a70a72f6acff67e20d144d627b0138611f49821
3
+ size 282847
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ streamlit
2
+ joblib
3
+ scikit-learn
train.csv ADDED
The diff for this file is too large to render. See raw diff