Update app.py
Browse files
app.py
CHANGED
|
@@ -36,6 +36,7 @@ df = pd.read_csv(data_path)
|
|
| 36 |
# Prepare features and target
|
| 37 |
X = df.drop(columns=['Salary'])
|
| 38 |
y = df['Salary']
|
|
|
|
| 39 |
|
| 40 |
# Pre-defined input choices
|
| 41 |
input_choices = {
|
|
|
|
| 36 |
# Prepare features and target
|
| 37 |
X = df.drop(columns=['Salary'])
|
| 38 |
y = df['Salary']
|
| 39 |
+
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1, random_state=123)
|
| 40 |
|
| 41 |
# Pre-defined input choices
|
| 42 |
input_choices = {
|