Skip to content

Commit 9dceaf3

Browse files
committed
remove deprecated action
1 parent bd40051 commit 9dceaf3

2 files changed

Lines changed: 8 additions & 56 deletions

File tree

.github/workflows/manual.yml

Lines changed: 0 additions & 46 deletions
This file was deleted.

starter/starter/ml/model.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,21 @@
11
from sklearn.metrics import fbeta_score, precision_score, recall_score
22

33

4-
# Optional: implement hyperparameter tuning.
54
def train_model(X_train, y_train):
65
"""
76
Trains a machine learning model and returns it.
87
98
Inputs
109
------
11-
X_train : np.array
10+
X_train : np.ndarray
1211
Training data.
13-
y_train : np.array
12+
y_train : np.ndarray
1413
Labels.
1514
Returns
1615
-------
17-
model
16+
model : RandomForestClassifier
1817
Trained machine learning model.
1918
"""
20-
2119
pass
2220

2321

@@ -27,9 +25,9 @@ def compute_model_metrics(y, preds):
2725
2826
Inputs
2927
------
30-
y : np.array
28+
y : np.ndarray
3129
Known labels, binarized.
32-
preds : np.array
30+
preds : np.ndarray
3331
Predicted labels, binarized.
3432
Returns
3533
-------
@@ -48,13 +46,13 @@ def inference(model, X):
4846
4947
Inputs
5048
------
51-
model : ???
49+
model : RandomForestClassifier
5250
Trained machine learning model.
53-
X : np.array
51+
X : np.ndarray
5452
Data used for prediction.
5553
Returns
5654
-------
57-
preds : np.array
55+
preds : np.ndarray
5856
Predictions from the model.
5957
"""
6058
pass

0 commit comments

Comments
 (0)