Adedeji Ayinde
11/07/2022, 6:47 PMclass model_wrapper:
@task
def fit_classifier(self, train: pd.DataFrame,
val: pd.DataFrame) -> JoblibSerializedFile:
# fetch the features and target columns from the train dataset
x = train[train.columns[1:]]
y = train[train.columns[0]]
# fetch the features and target columns from the validation dataset
eval_x = val[val.columns[1:]]
eval_y = val[val.columns[0]]
m = self.xgb_classifier()
# fit the model to the train data
m.fit(x, y, eval_set=[(eval_x, eval_y)])
working_dir = flytekit.current_context().working_directory
fname = os.path.join(working_dir, "model.joblib.dat")
joblib.dump(m, fname)
# return the serialized model
return JoblibSerializedFile(path=fname)
Error message
...
raise _user_exceptions.FlyteAssertion("Input was not specified for: {} of type {}".format(k, var.type))
flytekit.exceptions.user.FlyteAssertion: Input was not specified for: self of type simple: NONE
Evan Sadler
11/07/2022, 7:11 PMNiels Bantilan
11/07/2022, 11:24 PMEvan Sadler
11/08/2022, 12:20 PMNiels Bantilan
11/08/2022, 3:54 PM<mailto:niels@union.ai|niels@union.ai>