forked from EEBD_AI/wgz_forecast
25 lines
522 B
Python
25 lines
522 B
Python
import joblib
|
|
import numpy as np
|
|
|
|
|
|
def load_model(path:str):
|
|
gbm = joblib.load('./models/pv_pred.joblib')
|
|
return gbm
|
|
|
|
def pv_forecast(inputs: np.ndarray, model):
|
|
"""_summary_
|
|
|
|
Args:
|
|
inputs (np.ndarray): 输入序列
|
|
model (_type_): _description_
|
|
"""
|
|
out = model.predict([inputs])
|
|
return out
|
|
|
|
|
|
if __name__ == '__main__':
|
|
model = load_model('./models/pv_pred.joblib')
|
|
inputs = np.random.randn(24)
|
|
print(inputs.shape)
|
|
out = pv_forecast(inputs, model)
|
|
print(out) |