Python的机器学习库sklearn的代码实现

我老师对我们的要求是要实现朴素贝叶斯分类器，提交的文件要包含main.py和running.py还有train.py，这个用Python的sklearn库不是直接调用api再通过预估器等就可以实现了吗？怎么划分成3个Python文件呢？