benchmarks.wiki

benchmarks.wiki / Public workspace

BigCodeBench / BigCodeBench v0.1.0_hf fff51c41-5ab0-5669-a93e-f94ebfa102bb

Problem

Answer published by the source. Consult the official source to check your work against its answer.

complete prompt

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression

# Constants
FEATURES = ['feature '+str(i) for i in range(1, 11)]
TARGET = 'target'

def task_func(df):
    """
    Train a linear regression model on a given DataFrame.
    
    Parameters:
    df (DataFrame): The DataFrame with features and target.
    
    Returns:
    LinearRegression: The trained linear regression model.
    
    Requirements:
    - pandas
    - sklearn.model_selection.train_test_split
    - sklearn.linear_model.LinearRegression
    
    Raises:
    - The function will raise a ValueError is input df is not a DataFrame.

    Example:
    >>> import numpy as np
    >>> np.random.seed(0)
    >>> df = pd.DataFrame({'feature ' + str(i): np.random.rand(100) for i in range(1, 11)})
    >>> df['target'] = df.apply(lambda row: sum(row), axis=1)
    >>> model = task_func(df)
    >>> print(len(model.coef_))
    10
    """

instruct prompt

Train a linear regression model on a given DataFrame.
The function should raise the exception for: The function will raise a ValueError is input df is not a DataFrame.
The function should output with:
    LinearRegression: The trained linear regression model.
You should write self-contained code starting with:

Code

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
# Constants
FEATURES = ['feature '+str(i) for i in range(1, 11)]
TARGET = 'target'
def task_func(df):

code prompt

Code

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
# Constants
FEATURES = ['feature '+str(i) for i in range(1, 11)]
TARGET = 'target'
def task_func(df):

entry point

task_func

libs

  • pandas
  • sklearn

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

task_id
fff51c41-5ab0-5669-a93e-f94ebfa102bb
task_key
default--v0~2e1~2e0~5fhf--fff51c41-5ab0-5669-a93e-f94ebfa102bb
task_revision_id
1
config / split
default / v0.1.0_hf

Official source

initial import