Exemplos de Dataset.get_training_test_sets em Python

Exemplo n.º 1

 def evaluate(self, dataset: Dataset):
     _, x_test, _, y_test = dataset.get_training_test_sets()
     self._test(x_test, y_test, "test")
     print(self._model.score(x_test, y_test))

Exemplo n.º 2

 def fit(self, dataset: Dataset):
     x_train, _, y_train, _ = dataset.get_training_test_sets()
     self._model.fit(x_train, y_train)
     self._test(x_train, y_train, "train")
     mlflow.sklearn.log_model(self._model, "linear_model")