LucasSchettini
Segue um passo a passo:
1 - Criar seu ambiete vritual
mkdir teste
python -m venv teste/
2 - Carregar seu ambiente vritual
source teste/bin/activate
3 - Agora que seu ambiente virtual está carregado, você pode usar o pip para instalar seu pacote:
Se necessário atualize o pip:
pip install --upgrade pip
pip install scikit-learn
4 - Comece a programar adicionando este conteúdo no arquivo teste.py
vim teste.py
#!/usr/bin/env python
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error
X = [[1], [2], [3], [4], [5]]
y = [2.2, 2.8, 3.6, 4.5, 5.1]
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
model = LinearRegression()
model.fit(X_train, y_train)
y_pred = model.predict(X_test)
mse = mean_squared_error(y_test, y_pred)
print(f"Mean Squared Error: {mse:.2f}")
print(f"Coefficient: {model.coef_[0]:.2f}")
print(f"Intercept: {model.intercept_:.2f}")
5 - Dê permissão de execução
chmod +x teste.py
6 - Execute o script
./teste.py
Mean Squared Error: 0.02
Coefficient: 0.74
Intercept: 1.45
Só acho que se você esta começando, não deveria estar começando pelo módulo scikit-learn.