我在尝试将我的机器学习模型添加到Label Studio中。我尝试了以下命令:
cd label-studiopip install -e .cd label_studio/ml/examplespip install -r requirements.txt label-studio-ml init my_ml_backend --script label_studio/ml/examples/simple_text_classifier.py
在label_studio/ml/examples
目录下有以下文件:
simple_text_classifier.pydummy_model.pypytorch_transfer_learning.py
我遇到了以下错误:
ModuleNotFoundError: No module named 'simple_text_classifier'
我该如何在Label Studio中添加我的模型?
回答:
虽然回答得有点晚,但希望对其他人有帮助。我的解决方法是:在label_studio/ml/examples
文件夹中,尝试运行以下命令:
label-studio-ml init my_ml_backend --script simple_text_classifier.py
(我只是移除了脚本路径的其余部分)如果您收到提示需要添加--force
的错误,您可以选择添加--force
(这将覆盖现有的my_ml_backend
文件夹),或者简单地删除my_ml_backend
文件夹,然后再次运行相同的命令。
之后,您应该会看到以下响应:
Congratulations! ML Backend has been successfully initialized in ./my_ml_backendNow start it by using:label-studio-ml start ./my_ml_backend