The Submission Code Interface

Your submission needs to provide at least three components: imports, train(), and infer().

  1. imports: As with any script, if your solution contains dependencies on external packages make sure to import them. The system will automatically install your dependencies. Make sure that you are using only packages that are whitelisted here.

  2. train(): In the training phase the users will build the model and train it such that it can perform inferences on the testing data. The model must be saved in the resources directory.

  3. infer(): In the inference function the model trained in the previous step will be loaded and used to perform inferences on a data sample matching the characteristic of the training test.

A basic but functional submission is available here:

Code Signature

Since Python does not enforce types, you need to make sure that your code will expect these arguments.

def train(
    X_train: pandas.DataFrame,
    y_train: pandas.DataFrame,
    model_directory_path: str
) -> None
def infer(
    X_test: pandas.DataFrame,
    model_directory_path: str
) -> pandas.DataFrame

Last updated