Skip to content

Commit

Permalink
feat: change predict to forecast
Browse files Browse the repository at this point in the history
  • Loading branch information
Lopa10ko committed Mar 25, 2024
1 parent e6f19e7 commit a357726
Showing 1 changed file with 3 additions and 10 deletions.
13 changes: 3 additions & 10 deletions frameworks/FEDOT/exec_ts.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,14 +53,7 @@ def run(dataset, config):
)

test_sub_df = test_df[test_df[id_column] == label].drop(columns=[id_column], axis=1)
test_series = test_sub_df[dataset.target].to_numpy()
test_input = InputData(
idx=np.arange(len(test_series)),
features=train_series,
target=test_series,
task=task,
data_type=DataTypesEnum.ts
)
horizon = len(test_sub_df[dataset.target])

fedot = Fedot(
problem=TaskTypesEnum.ts_forecasting.value,
Expand All @@ -78,10 +71,10 @@ def run(dataset, config):

with Timer() as predict:
try:
prediction = fedot.predict(test_input)
prediction = fedot.forecast(train_input, horizon=horizon)
except Exception as e:
log.info('Pipeline crashed. Using no-op forecasting')
prediction = np.full(len(test_series), train_series[-1])
prediction = np.full(horizon, train_series[-1])

predict_duration += predict.duration

Expand Down

0 comments on commit a357726

Please sign in to comment.