def $\_\_$init$\_\_($self$)$:

# Your code starts after this line

# Your code ends before this line

def train$($self$,$ data, column, values$)$:

# Your code starts after this line

# Your code ends before this line

def predict$($self$)$:

# Your code starts after this line

# Your code ends before this line

import pandas as pd

data $=$ pd$.$read$\_$csv$("$data$/$da$-$week$1.$csv$")$

model$1=$ model$()$

model$1.$train$($data$,\text{'}$X$\text{'},[\text{'}$A$\text{'},\text{'}$B$\text{'},\text{'}$C$\text{'},\text{'}$D$\text{'}])$

print$(\text{'}$Model $1$ prediction: $\text{'}+$ model$1.$predict$())$

model$2=$ model$()$

model$2.$train$($data$,\text{'}$Y$\text{'},$ range$(1,11))$

print$(\text{'}$Model $2$ prediction: $\text{'}+$str$($model$2.$predict$()))$