Source code for ads.dataset.regression_dataset
#!/usr/bin/env python
# -*- coding: utf-8; -*-
# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
from ads.dataset.dataset_with_target import ADSDatasetWithTarget
[docs]
class RegressionDataset(ADSDatasetWithTarget):
def __init__(self, df, sampled_df, target, target_type, shape, **kwargs):
ADSDatasetWithTarget.__init__(
self, df=df, sampled_df=sampled_df, target=target, target_type=target_type, shape=shape, **kwargs
)