codenet-mlm.yaml

# Copyright 2021 The MLX Contributors
#
# SPDX-License-Identifier: Apache-2.0

id: 'project-codenet-mlm'
name: 'CodeNet Masked Language Model'
description: 'Experiment investigates whether a popular attention model to construct a masked language model (MLM) can be used for source code instead of natural language sentences'
metadata:
  annotations:
    platform: 'OpenSource'
implementation:
  github:
    source: 'https://github.com/machine-learning-exchange/katalog/blob/main/notebook-samples/src/codenet/Project_CodeNet_MLM.ipynb'
    requirements: 'keras==2.7.0,matplotlib==3.5.1,pandas==1.3.5,tensorflow==2.7.0,Jinja2==3.0.3'
    image: 'tensorflow/tensorflow:2.7.0'