mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-30 11:26:23 +02:00
delete feature.csv, store.csv, test.csv.
This commit is contained in:
parent
fc40174802
commit
6652aa09ce
1 changed files with 3 additions and 7 deletions
|
|
@ -10,22 +10,17 @@ DATA_DIR = "examples/mi/data/WalmartSalesForecast2"
|
|||
SALES_FORECAST_REQ = f"""
|
||||
# Goal
|
||||
Train a model to predict sales for each department in every store (split the last 40 weeks records as validation dataset,
|
||||
the others is train dataset), include plot sales trends, holiday effects, distribution of sales across stores/departments,
|
||||
using box on the train dataset, print metric and plot scatter plots of groud truth and predictions on validation data.
|
||||
save predictions on test data.
|
||||
the others is train dataset), include plot sales trends, print metric and plot scatter plots of
|
||||
groud truth and predictions on validation data.
|
||||
|
||||
# Datasets Available
|
||||
- train_data: {DATA_DIR}/train.csv
|
||||
- test_data: {DATA_DIR}/test.csv, no label data.
|
||||
- additional data: {DATA_DIR}/features.csv
|
||||
- stores data: {DATA_DIR}/stores.csv
|
||||
|
||||
# Metric
|
||||
The metric of the competition is weighted mean absolute error (WMAE) for test data.
|
||||
|
||||
# Notice
|
||||
- *print* key variables to get more information for next task step.
|
||||
- Only When you fit the model, make the DataFrame.dtypes to be int, float or bool, and drop date column.
|
||||
"""
|
||||
|
||||
requirements = {"wine": WINE_REQ, "sales_forecast": SALES_FORECAST_REQ}
|
||||
|
|
@ -36,6 +31,7 @@ async def main(auto_run: bool = True, use_case: str = "wine"):
|
|||
if use_case == "wine":
|
||||
requirement = requirements[use_case]
|
||||
else:
|
||||
mi.use_tools = True
|
||||
assert DATA_DIR != "your/path/to/data", f"Please set DATA_DIR for the use_case: {use_case}!"
|
||||
requirement = requirements[use_case]
|
||||
await mi.run(requirement)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue