19 lines
526 B
Python
19 lines
526 B
Python
from abc import ABC, abstractmethod
|
|
import pandas as pd
|
|
|
|
class IDataSource(ABC):
|
|
"""
|
|
数据源接口，定义了获取训练数据的标准方法。
|
|
"""
|
|
@abstractmethod
|
|
def get_data(self, **filters) -> pd.DataFrame:
|
|
"""
|
|
根据指定的筛选条件获取数据。
|
|
|
|
Args:
|
|
**filters: 一个包含筛选条件的字典，例如 store_ids=['S001'], product_ids=['P001']。
|
|
|
|
Returns:
|
|
一个包含所需数据的 pandas DataFrame。
|
|
"""
|
|
pass |