(self)
| 49 | self.load_qlib_data() |
| 50 | |
| 51 | def load_qlib_data(self): |
| 52 | instruments = D.instruments(market="all") |
| 53 | instrument_list = D.list_instruments(instruments=instruments, as_list=True, freq=self.freq) |
| 54 | required_fields = ["$open", "$close", "$low", "$high", "$volume", "$factor"] |
| 55 | for instrument in instrument_list: |
| 56 | df = D.features([instrument], required_fields, freq=self.freq) |
| 57 | df.rename( |
| 58 | columns={ |
| 59 | "$open": "open", |
| 60 | "$close": "close", |
| 61 | "$low": "low", |
| 62 | "$high": "high", |
| 63 | "$volume": "volume", |
| 64 | "$factor": "factor", |
| 65 | }, |
| 66 | inplace=True, |
| 67 | ) |
| 68 | self.data[instrument] = df |
| 69 | print(df) |
| 70 | |
| 71 | def check_missing_data(self) -> Optional[pd.DataFrame]: |
| 72 | """Check if any data is missing in the DataFrame.""" |
no test coverage detected