[英]Gitlab CI/CD error Machine Learning Python
我在 Modelrf 类中编写了一个 RandomForest 回归模型。 它是一个名为 RandomForest.py 的单独 python 文件
RandomForest.py 的代码:
from sklearn.ensemble import RandomForestRegressor
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
import numpy as np
import pandas as pd
class Modelrf():
def __init__(self, train = "train.csv", test = "test.csv"):
self.X_train = pd.read_csv(train)
self.X_test = pd.read_csv(test)
self.linear_reg = LinearRegression()
self.random_forest = RandomForestRegressor()
def split(self):
self.X_train.dropna(axis=0, subset=['salary'], inplace=True)
self.X_test.dropna(axis=0, subset=['salary'], inplace=True)
self.y_train = self.X_train.final_hourly_fee
self.y_test = self.X_test.final_hourly_fee
def fit(self):
self.model = self.random_forest.fit(self.X_train, self.y_train)
def predict(self):
self.result = self.random_forest.predict(self.X_test)
return self.result
if __name__ == '__main__':
model_instance = Modelrf()
model_instance.split()
model_instance.fit()
model_instance.predict()
print(model_instance.result)
print("Accuracy: ", model_instance.model.score(model_instance.X_test, model_instance.y_test))
output = pd.DataFrame({'Id': model_instance.X_test.index,'Y Original': model_instance.y_test, 'Y predicted':model_instance.result})
output.to_csv('outputTest.txt', index=False)
现在我已经将 Modelrf 类导入到 main.py
main.py 的代码:
import pandas as pd
import numpy as np
from RandomForest import Modelrf
from RandomForest import X_test,y_test,result
if __name__ == '__main__':
model_instance = Modelrf()
model_instance.split()
model_instance.fit()
model_instance.predict()
print(model_instance.result)
print("Accuracy: ", model_instance.model.score(model_instance.X_test, model_instance.y_test))
output = pd.DataFrame({'Id': model_instance.X_test.index,'Y Original': model_instance.y_test, 'Y predicted':model_instance.result})
output.to_csv('outputTest.txt', index=False)
我想在 gitlab 上启用 CI/CD 如何编写我的“.gitlab-ci.yml”文件代码?
如果我正确理解这种情况,您需要转到您的存储库:设置 > CI/CD > Runners 然后您需要决定是使用私有运行器还是共享运行器:
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.