note on adding more input variables.

carpentries-incubator · Aug 23, 2022 · 15659f8 · 15659f8
1 parent 58b5056
commit 15659f8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/_episodes/04-modelling.md b/_episodes/04-modelling.md
@@ -39,6 +39,7 @@ from sklearn.linear_model import LinearRegression
 reg = LinearRegression()
 
 # use a single feature (apache score)
+# note: remove the reshape if fitting to >1 input variable
 X = cohort.apachescore.values.reshape(-1, 1)
 y = cohort.actualhospitalmortality_enc.values
 

diff --git a/_episodes/05-validation.md b/_episodes/05-validation.md
@@ -50,6 +50,7 @@ y = cohort[outcome]
 x_train, x_test, y_train, y_test = train_test_split(X, y, train_size = 0.7, random_state = 42)
 
 # restructure data for input into model
+# note: remove the reshape if fitting to >1 input variable
 x_train = x_train.values.reshape(-1, 1)
 y_train = y_train.values.ravel()
 x_test = x_test.values.reshape(-1, 1)

diff --git a/_episodes/06-evaluation.md b/_episodes/06-evaluation.md
@@ -44,6 +44,7 @@ y = cohort[outcome]
 x_train, x_test, y_train, y_test = train_test_split(X, y, train_size = 0.7, random_state =  42)
 
 # restructure data for input into model
+# note: remove the reshape if fitting to >1 input variable
 x_train = x_train.values.reshape(-1, 1)
 y_train = y_train.values.ravel()
 x_test = x_test.values.reshape(-1, 1)

diff --git a/_episodes/07-bootstrapping.md b/_episodes/07-bootstrapping.md
@@ -55,6 +55,7 @@ y = cohort[outcome]
 x_train, x_test, y_train, y_test = train_test_split(X, y, train_size = 0.7, random_state = 42)
 
 # restructure data for input into model
+# note: remove the reshape if fitting to >1 input variable
 x_train = x_train.values.reshape(-1, 1)
 y_train = y_train.values.ravel()
 x_test = x_test.values.reshape(-1, 1)